1 環(huán)境搭建

1.1 在官方倉(cāng)庫(kù)的給定的使用python=3.9版本，則使用conda創(chuàng)建對(duì)應(yīng)虛擬環(huán)境。

conda create -n yolov10 python=3.9

1.2 切換到對(duì)應(yīng)虛擬環(huán)境

conda activate yolov10

1.3 在指定目錄下克隆yolov10官方倉(cāng)庫(kù)代碼

git clone https://github.com/THU-MIG/yolov10.git

1.4 進(jìn)入對(duì)應(yīng)本地yolov10目錄下安裝對(duì)應(yīng)依賴項(xiàng)

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/

如果使用 pip install -r requirements.txt 命令，可能會(huì)遇到錯(cuò)誤

WARNING: Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None)) after connection broken by 'ConnectTimeoutError(<pip._vendor.urllib3.connection.HTTPSConnection object at 0x000001A7EADD6CD0>, 'Connection to files.pythonhosted.org timed out. (connect timeout=15)')': /packages/48/f4/d0b61525a3d3db78636f1937d1bc24cbb39abc57484a545b72b6ab35c114/torch-2.0.1-cp39-cp39-win_amd64.whl
WARNING: Retrying (Retry(total=3, connect=None, read=None, redirect=None, status=None)) after connection broken by 'ConnectTimeoutError(<pip._vendor.urllib3.connection.HTTPSConnection object at 0x000001A7EADD6CA0>, 'Connection to files.pythonhosted.org timed out. (connect timeout=15)')': /packages/48/f4/d0b61525a3d3db78636f1937d1bc24cbb39abc57484a545b72b6ab35c114/torch-2.0.1-cp39-cp39-win_amd64.whl
WARNING: Retrying (Retry(total=2, connect=None, read=None, redirect=None, status=None)) after connection broken by 'ConnectTimeoutError(<pip._vendor.urllib3.connection.HTTPSConnection object at 0x000001A7EB200400>, 'Connection to files.pythonhosted.org timed out. (connect timeout=15)')': /packages/48/f4/d0b61525a3d3db78636f1937d1bc24cbb39abc57484a545b72b6ab35c114/torch-2.0.1-cp39-cp39-win_amd64.whl
WARNING: Retrying (Retry(total=1, connect=None, read=None, redirect=None, status=None)) after connection broken by 'ConnectTimeoutError(<pip._vendor.urllib3.connection.HTTPSConnection object at 0x000001A7EB1F1F10>, 'Connection to files.pythonhosted.org timed out. (connect timeout=15)')': /packages/48/f4/d0b61525a3d3db78636f1937d1bc24cbb39abc57484a545b72b6ab35c114/torch-2.0.1-cp39-cp39-win_amd64.whl
WARNING: Retrying (Retry(total=0, connect=None, read=None, redirect=None, status=None)) after connection broken by 'ConnectTimeoutError(<pip._vendor.urllib3.connection.HTTPSConnection object at 0x000001A7EB1F15B0>, 'Connection to files.pythonhosted.org timed out. (connect timeout=15)')': /packages/48/f4/d0b61525a3d3db78636f1937d1bc24cbb39abc57484a545b72b6ab35c114/torch-2.0.1-cp39-cp39-win_amd64.whl
ERROR: Could not install packages due to an OSError: HTTPSConnectionPool(host='files.pythonhosted.org', port=443): Max retries exceeded with url: /packages/48/f4/d0b61525a3d3db78636f1937d1bc24cbb39abc57484a545b72b6ab35c114/torch-2.0.1-cp39-cp39-win_amd64.whl (Caused by ConnectTimeoutError(<pip._vendor.urllib3.connection.HTTPSConnection object at 0x000001A7ECD869A0>, 'Connection to files.pythonhosted.org timed out. (connect timeout=15)'))

1.5 安裝對(duì)應(yīng)的包

pip install -e .

2 模型訓(xùn)練及測(cè)試效果

2.1 在Roboflow下載數(shù)據(jù)集并導(dǎo)出為YOLO的數(shù)據(jù)集（? 或者在Roboflow制作自己的數(shù)據(jù)集并導(dǎo)出為YOLO的數(shù)據(jù)集）。參考Roboflow 使用詳解：數(shù)據(jù)集標(biāo)注、訓(xùn)練及下載

2.3 把data.yaml復(fù)制到C:\Users\WYX\Desktop\code\yolov10-main\ultralytics\cfg\datasets目錄下，重命名為VOC_new.yaml，并修改相應(yīng)的配置。

2.4 模型訓(xùn)練?

需要修改C:\Users\WYX\Desktop\code\yolov10-main\ultralytics\models\yolov10\train.py。

訓(xùn)練

yolo detect train data=VOC_new.yaml model=yolov10n.yaml epochs=500 batch=32 imgsz=640 device=cpu

2.5 模型驗(yàn)證?

yolo val model=jameslahm/yolov10n data=VOC_new.yaml batch=32 

2.6 預(yù)測(cè)并導(dǎo)出

yolo export model=jameslahm/yolov10n format=onnx opset=13 simplify