the files need to be recognised

rohitrango · Oct 28, 2020 · fb091c0 · fb091c0
1 parent a11d4e0
commit fb091c0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/__init__.py b/src/__init__.py
@@ -1,4 +1,4 @@
-from estimate_watermark import *
-from preprocess import *
-from image_crawler import *
-from watermark_reconstruct import *
+from .estimate_watermark import *
+from .preprocess import *
+from .image_crawler import *
+from .watermark_reconstruct import *
diff --git a/src/watermark_reconstruct.py b/src/watermark_reconstruct.py
@@ -4,8 +4,8 @@
 import scipy
 from scipy.sparse import *
 from scipy.sparse import linalg
-from estimate_watermark import *
-from closed_form_matting import *
+from .estimate_watermark import *
+from .closed_form_matting import *
 from numpy import nan, isnan
 
 def get_cropped_images(foldername, num_images, start, end, shape):
@@ -302,4 +302,4 @@ def changeContrastImage(J, I):
     cI2 = I[-1,-1, :]
 
     I_m = cJ1 + (I-cI1)/(cI2-cI1)*(cJ2-cJ1)
-    return I_m
+    return I_m