yolov5训练自定义模型——环境配置
安装CUDA
安装cudnn
安装yolov5
安装yolov5依赖
遇到问题:在跑测试安装代码时
python detect.py --source ./data/images/ --weights weights/yolov5s.pt --conf-thres 0.4
遇到错误
(pytorch) C:\Program Files\yolo\yolov5>python detect.py --source ./data/images/ --weights weights/yolov5s.pt --conf-thres 0.4
Downloading https://ultralytics.com/assets/Arial.ttf to C:\Users\ShiYa\AppData\Roaming\Ultralytics\Arial.ttf...
detect: weights=['weights/yolov5s.pt'], source=./data/images/, data=data\coco128.yaml, imgsz=[640, 640], conf_thres=0.4, iou_thres=0.45, max_det=1000, device=, view_img=False, save_txt=False, save_conf=False, save_crop=False, nosave=False, classes=None, agnostic_nms=False, augment=False, visualize=False, update=False, project=runs\detect, name=exp, exist_ok&