Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -152,9 +152,9 @@ class Dataset(BaseDataset):
|
|
152 |
image = cv2.imread(self.images_fps)
|
153 |
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
|
154 |
mask = cv2.imread(self.masks_fps, 0)
|
155 |
-
|
156 |
-
|
157 |
-
|
158 |
masks = [(mask == v) for v in self.class_values]
|
159 |
mask = np.stack(masks, axis=-1).astype('float')
|
160 |
|
@@ -221,7 +221,8 @@ def model_infer(img_name):
|
|
221 |
image, mask = input
|
222 |
|
223 |
image_vis = image.transpose(1, 2, 0)
|
224 |
-
gt_mask = mask[0]
|
|
|
225 |
pr_mask = (output[0].numpy() > threshold).astype('uint8')[0]
|
226 |
i = i + 1
|
227 |
if i >= break_at:
|
|
|
152 |
image = cv2.imread(self.images_fps)
|
153 |
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
|
154 |
mask = cv2.imread(self.masks_fps, 0)
|
155 |
+
mask[np.where(mask < 8)] = 0
|
156 |
+
mask[np.where(mask > 8)] = 255
|
157 |
+
extract certain classes from mask (e.g. polyp)
|
158 |
masks = [(mask == v) for v in self.class_values]
|
159 |
mask = np.stack(masks, axis=-1).astype('float')
|
160 |
|
|
|
221 |
image, mask = input
|
222 |
|
223 |
image_vis = image.transpose(1, 2, 0)
|
224 |
+
gt_mask = mask[0]
|
225 |
+
#gt_mask = mask[0].astype('uint8')
|
226 |
pr_mask = (output[0].numpy() > threshold).astype('uint8')[0]
|
227 |
i = i + 1
|
228 |
if i >= break_at:
|