File size: 369 Bytes
1e4b791
 
 
 
 
 
 
1
2
3
4
5
6
7
8
# train yolov5s on custom data for 100 epochs
# time its performance
import os

os.environ['KMP_DUPLICATE_LIB_OK']='True'
python test.py
# python train.py --img 208 --batch 16 --epochs 100 --data ./data.yaml --cfg ./models/custom_yolov5s.yaml --weights 'C:\Users\Samuel\Downloads\yolov5\runs\train\yolov5s_results19\weights\best' --name yolov5s_results  --cache