File size: 666 Bytes
ecebdee 91251f4 ecebdee 0f58fe6 ecebdee 91251f4 0f58fe6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | ---
tags:
- fastai
---
# Model card
## Model description
A crude CV model to classify retinal images. I started with a pretrained model (resnet18) and tuned it with a set of retinal images.
## Intended uses & limitations
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.
## Training and evaluation data
About 400 retinal images were downloaded from <a href="http://cecas.clemson.edu/~ahoover/stare/">STARE</a>.
The model was tuned with 50 epochs. accuracy-multi: 91.4%; valid-loss: 26.0%; train-loss: 26.0%
|