glaucoma_screening_confidence

Sleeping

pamixsun commited on Aug 22, 2023

Commit

a21a9d8

•

1 Parent(s): a504466

Update glaucoma.py

Files changed (1) hide show

glaucoma.py CHANGED Viewed

@@ -85,7 +85,7 @@ class GlaucomaModel(object):
         try:
             vcdr = simple_vcdr(disc_cup)
         except:
-            vcdr = '[None]'
         _, disc_cup_image = add_mask(image, disc_cup, [0, 1, 2], [[0, 0, 0], [0, 255, 0], [255, 0, 0]], 0.2)
         return disease_idx, disc_cup_image, cam, vcdr

         try:
             vcdr = simple_vcdr(disc_cup)
         except:
+            vcdr = np.nan
         _, disc_cup_image = add_mask(image, disc_cup, [0, 1, 2], [[0, 0, 0], [0, 255, 0], [255, 0, 0]], 0.2)
         return disease_idx, disc_cup_image, cam, vcdr