Downloads · 30 days
11
13% of all-time downloads
mohammedelabbas/speyeria-classifier
speyeria-classifier is a image classification model from mohammedelabbas. Use it when you need a label for an image. It is set up for pytorch. The card lists the license as mit.
This repository contains a ResNet-50 based PyTorch classifier for identifying Speyeria species from images.
Downloads · 30 days
11
13% of all-time downloads
All-time downloads
86
Public
Parameters
23.6M
94.4 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors94.4 MB · 100%
From the Hugging Face model README
This repository contains a ResNet-50 based PyTorch classifier for identifying Speyeria species from images.
The model predicts the following species (short labels are used in config.json):
Use model.py to instantiate the architecture and load weights:
from safetensors.torch import load_file
from model import create_model
model = create_model(num_classes=16)
state_dict = load_file("model.safetensors")
model.load_state_dict(state_dict)
model.eval()
config.json contains id2label/label2id mappings..safetensors format for safer deserialization.