Spaces:
Running
Running
Update glaucoma.py
Browse files- glaucoma.py +1 -1
glaucoma.py
CHANGED
@@ -86,7 +86,7 @@ class GlaucomaModel(object):
|
|
86 |
vcdr = simple_vcdr(disc_cup)
|
87 |
except:
|
88 |
vcdr = np.nan
|
89 |
-
_, disc_cup_image = add_mask(image, disc_cup, [
|
90 |
|
91 |
return disease_idx, disc_cup_image, cam, vcdr
|
92 |
|
|
|
86 |
vcdr = simple_vcdr(disc_cup)
|
87 |
except:
|
88 |
vcdr = np.nan
|
89 |
+
_, disc_cup_image = add_mask(image, disc_cup, [1, 2], [[0, 255, 0], [255, 0, 0]], 0.2)
|
90 |
|
91 |
return disease_idx, disc_cup_image, cam, vcdr
|
92 |
|