no message

Author: sniklaus

correlation/correlation.py

@@ -1,9 +1,8 @@
 #!/usr/bin/env python
 
-import torch
-
 import cupy
 import re
+import torch
 
 kernel_Correlation_rearrange = '''
     extern "C" __global__ void kernel_Correlation_rearrange(

run.py

@@ -1,14 +1,12 @@
 #!/usr/bin/env python
 
-import torch
-
 import getopt
 import math
 import numpy
-import os
 import PIL
 import PIL.Image
 import sys
+import torch
 
 try:
     from .correlation import correlation # the custom cost volume layer