Update README.md
Browse files
README.md
CHANGED
@@ -6,12 +6,12 @@ tags:
|
|
6 |
# Model card
|
7 |
|
8 |
## Model description
|
9 |
-
A crude CV to classify retinal images. I started with a pretrained model (resnet18) and tuned it with a set of retinal images.
|
10 |
|
11 |
## Intended uses & limitations
|
12 |
Limitations stem primarily from a naïve use of the training data. Imbalances exist in the number of training images associated with each condition. In the future, a weighting function could be applied.
|
13 |
|
14 |
## Training and evaluation data
|
15 |
-
About 400 retinal images were downloaded from
|
16 |
|
17 |
The model was tuned with 50 epochs. accuracy-multi: 91.4%; valid-loss: 26.0%; train-loss: 26.0%
|
|
|
6 |
# Model card
|
7 |
|
8 |
## Model description
|
9 |
+
A crude CV model to classify retinal images. I started with a pretrained model (resnet18) and tuned it with a set of retinal images.
|
10 |
|
11 |
## Intended uses & limitations
|
12 |
Limitations stem primarily from a naïve use of the training data. Imbalances exist in the number of training images associated with each condition. In the future, a weighting function could be applied.
|
13 |
|
14 |
## Training and evaluation data
|
15 |
+
About 400 retinal images were downloaded from <a href="http://cecas.clemson.edu/~ahoover/stare/">STARE</a>.
|
16 |
|
17 |
The model was tuned with 50 epochs. accuracy-multi: 91.4%; valid-loss: 26.0%; train-loss: 26.0%
|