Update predict.py

jianlin-cheng · Dec 14, 2023 · 4e9df1a · 4e9df1a
1 parent 8da3fc6
commit 4e9df1a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/predict.py b/predict.py
@@ -78,7 +78,6 @@ def make_predictions(model, image_path):
         # image = image.T
         # image = np.rot90(image)
         image = cv2.imread(image_path, 0)
-        image = denoise_jpg_image(image)
         mask = cv2.imread(mask_path, 0)
         height, width = image.shape
         orig_image = copy.deepcopy(image)
@@ -137,4 +136,4 @@ def make_predictions(model, image_path):
 images_path = list(glob.glob(f"{config.test_dataset_path}/{config.empiar_id}/images/*.jpg"))
 
 for i in range(0, len(images_path), 1):
-	make_predictions(model, images_path[i])
+	make_predictions(model, images_path[i])