| | --- |
| | 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% |
| |
|
