Skip to content

Commit 9ea1a08

Browse files
author
Yifan Peng
committed
fix bugs
1 parent 62172c2 commit 9ea1a08

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

deepseenet/deepseenet_simplified.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def has_intermediate_drusen(score):
4747
return 5 if score >= 5 else score
4848

4949

50-
class EyesNetSimplifiedScore(object):
50+
class DeepSeeNetSimplifiedScore(object):
5151
def __init__(self, drusen_model='areds', pigment_model='areds', advanced_amd_model='areds'):
5252
"""
5353
Args:

examples/predict_simplified_score.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
import numpy as np
1717
from keras.preprocessing import image
1818

19-
from deepseenet import eyesnet_simplified
19+
from deepseenet import deepseenet_simplified
2020
from deepseenet.utils import crop2square, pick_device
2121

2222

@@ -49,6 +49,6 @@ def preprocess_image(image_path, target_size=(224, 224)):
4949
advanced_amd_model = argv['-a'] if '-a' in argv else None
5050

5151
pick_device()
52-
clf = eyesnet_simplified.EyesNetSimplifiedScore(drusen_model, pigment_model, advanced_amd_model)
52+
clf = deepseenet_simplified.DeepSeeNetSimplifiedScore(drusen_model, pigment_model, advanced_amd_model)
5353
score = clf.predict(argv['<left_eye_image>'], argv['<right_eye_image>'], verbose=1)
5454
print('The simplified score:', score)

0 commit comments

Comments
 (0)