Update README.md
Browse files
README.md
CHANGED
|
@@ -30,4 +30,10 @@ img = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
|
|
| 30 |
|
| 31 |
# perform inference
|
| 32 |
results = model(img)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
```
|
|
|
|
| 30 |
|
| 31 |
# perform inference
|
| 32 |
results = model(img)
|
| 33 |
+
```
|
| 34 |
+
|
| 35 |
+
- Finetune the model on your custom dataset:
|
| 36 |
+
|
| 37 |
+
```bash
|
| 38 |
+
yolov5 classify train --img 128 --data mnist2560 --model fcakyon/yolov5n-cls-v7.0 --epochs 1 --device cpu
|
| 39 |
```
|