Downloads · 30 days
11
0% of all-time downloads
yolov5n-cls-v7.0
fcakyon/yolov5n-cls-v7.0
yolov5n-cls-v7.0 is a image classification model from fcakyon. Use it when you need a label for an image. It is set up for transformers. The card lists the license as gpl-3.0.
All-time downloads
4.4K
Public
Repo size
5.1 MB
Likes
2
Public
Repo makeup
Click a slice to open those files.
.pt5.1 MB · 100%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- gpl-3.0
- Access
- Public
- Created
- Dec 13, 2022
- Updated
- Dec 20, 2022
- SHA
- 3c8ab229
Model card
From the Hugging Face model README
How to use
- Install yolov5:
pip install -U yolov5
- Load model and perform prediction:
import yolov5
# load model
model = yolov5.load('fcakyon/yolov5n-cls-v7.0')
# set image
img = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
# perform inference
results = model(img)
- Finetune the model on your custom dataset:
yolov5 classify train --img 128 --data mnist2560 --model fcakyon/yolov5n-cls-v7.0 --epochs 1 --device cpu
- Task
- Image Classification
- Library
- transformers
- License
- gpl-3.0
- Created
- Dec 13, 2022
- Updated
- Dec 20, 2022