Downloads · 30 days
10
1% of all-time downloads
nateraw/resnet50
resnet50 is a image classification model from nateraw. Use it when you need a label for an image. It is set up for transformers.
Downloads · 30 days
10
1% of all-time downloads
All-time downloads
1.3K
Public
Repo size
205 MB
Likes
0
Public
Click a slice to open those files.
.bin103 MB · 100%
From the Hugging Face model README
pip install modelz
from modelz import ResnetModel
model = ResnetModel.from_pretrained('nateraw/resnet50')
ex_input = torch.rand(4, 3, 224, 224)
out = model(ex_input)