the files need to be recognised

Author: CodeSaint98

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 *

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