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 STARE.
The model was tuned with 50 epochs. accuracy-multi: 91.4%; valid-loss: 26.0%; train-loss: 26.0%