资料:
https://github.com/ultralytics/yolov5/issues/36
本地下载好仓库/data/xiedong/eff_train/yolov5-master
给权重路径:/data/xiedong/eff_train/yolov5-master/yolov5n640/weights/best.pt
显卡跑就加这两句:
device=‘cuda:0’,
force_reload=True
cpu跑就别加这两句。
代码: