iicd/RAFT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added cuda extension for efficent implementation

Browse Source
This commit is contained in:
Zach Teed
2020-08-22 18:49:24 -06:00
parent 5b1f510d6b
commit c86b3dc8f3
13 changed files with 519 additions and 191 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
core/utils/augmentor.py
View File

@@ -4,8 +4,10 @@ import math
from PIL import Image
import cv2
import torch
cv2.setNumThreads(0)
cv2.ocl.setUseOpenCL(False)
import torch
from torchvision.transforms import ColorJitter
import torch.nn.functional as F