requirements and gitignore

.gitignore

@@ -0,0 +1,4 @@
+venv
+__pycache__
+.Ds_Store
+train

requirements.txt

@@ -0,0 +1,19 @@
+contourpy==1.3.1
+cycler==0.12.1
+fonttools==4.55.3
+imageio==2.36.1
+kiwisolver==1.4.7
+lazy_loader==0.4
+matplotlib==3.10.0
+networkx==3.4.2
+numpy==2.2.0
+opencv-python==4.10.0.84
+packaging==24.2
+pillow==11.0.0
+pyparsing==3.2.0
+python-dateutil==2.9.0.post0
+scikit-image==0.25.0
+scipy==1.14.1
+six==1.17.0
+tifffile==2024.12.12
+tqdm==4.67.1

train.py

@@ -7,7 +7,8 @@
 from skimage.transform import resize
 import network
 import os
-
+import cv2
+import glob
 # Helper Functions
 def get_corrupted_input(input, corruption_level):
     corrupted = np.copy(input)
@@ -34,7 +35,7 @@ def plot(data, test, predicted, figsize=(5, 6)):
     count=0
     file_count=0
     for d in split(data, 4):
-        if not len(d)is 1: 
+        if len(d) != 1: 
             fig, axarr = plt.subplots(len(d), 3)
             for i in range(len(d)):
                 if i==0:
@@ -87,9 +88,6 @@ def preprocessing(img, w=128, h=128):
 
 def main():
     # Load data
-
-    import cv2
-    import glob
     img_dir = "train/" # Enter Directory of all images 
     data_path = os.path.join(img_dir,'*g')
     files = glob.glob(data_path)