Spaces:
Runtime error
Runtime error
Update glaucoma.py
Browse files- glaucoma.py +4 -1
glaucoma.py
CHANGED
|
@@ -82,7 +82,10 @@ class GlaucomaModel(object):
|
|
| 82 |
cam = self.cam.get_cam(image, disease_idx)
|
| 83 |
cam = cv2.resize(cam, image_shape[::-1])
|
| 84 |
disc_cup = self.optic_disc_cup_pred(image)
|
| 85 |
-
|
|
|
|
|
|
|
|
|
|
| 86 |
_, disc_cup_image = add_mask(image, disc_cup, [0, 1, 2], [[0, 0, 0], [0, 255, 0], [255, 0, 0]], 0.2)
|
| 87 |
|
| 88 |
return disease_idx, disc_cup_image, cam, vcdr
|
|
|
|
| 82 |
cam = self.cam.get_cam(image, disease_idx)
|
| 83 |
cam = cv2.resize(cam, image_shape[::-1])
|
| 84 |
disc_cup = self.optic_disc_cup_pred(image)
|
| 85 |
+
try:
|
| 86 |
+
vcdr = simple_vcdr(disc_cup)
|
| 87 |
+
except:
|
| 88 |
+
vcdr = '[None]'
|
| 89 |
_, disc_cup_image = add_mask(image, disc_cup, [0, 1, 2], [[0, 0, 0], [0, 255, 0], [255, 0, 0]], 0.2)
|
| 90 |
|
| 91 |
return disease_idx, disc_cup_image, cam, vcdr
|