pamixsun commited on
Commit
053a66b
1 Parent(s): a21a9d8

Update glaucoma.py

Browse files
Files changed (1) hide show
  1. 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, [0, 1, 2], [[0, 0, 0], [0, 255, 0], [255, 0, 0]], 0.2)
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