From 8da3fc61c28ce857ce22e989ffecd4e27cccaca7 Mon Sep 17 00:00:00 2001 From: rajangyawali Date: Wed, 13 Dec 2023 07:41:36 -0600 Subject: [PATCH] Updated Predicted --- predict.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/predict.py b/predict.py index 0f03be1..3206509 100644 --- a/predict.py +++ b/predict.py @@ -1,7 +1,7 @@ # Code for making predictions on individual micrographs import copy -from utils.denoise import denoise +from utils.denoise import denoise, denoise_jpg_image import config import matplotlib.pyplot as plt import numpy as np @@ -78,6 +78,7 @@ 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)