模型训练:
参考这篇:
YOLOv10训练自己的数据集(从代码下载到实例测试)-CSDN博客
跟着这个步骤实现训练,
模型测试:
from ultralytics import YOLOv10
yolo = YOLOv10("runs/detect/train/weights/best.pt",task="detect")
result = yolo(source=r"E:\yolov10-main\datasets\1\test",save=True,save_conf=True,save_txt=True,name='output')
source 换成自己的测试集路径
遇到的问题:
1.ImportError: cannot import name 'metadata' from 'importlib'
虚拟环境是python=3.7的 不太支持 换成python=3.8 的解决问题
2.OSError: [WinError 1455] 页面文件太小,无法完成操作。
解决办法: 修改batch 的大小