環(huán)境配置見yolov5/yolov3 pytorch環(huán)境配置gpu+windows11+anaconda+pycharm+RTX3050 筆記

正確的訓(xùn)練步驟見（windows）yolov5/yolov3訓(xùn)練自己的數(shù)據(jù)集，win11 gpu,pycharm,RTX3050筆記

數(shù)據(jù)集文件分布

建議將JPEGImages改為images,否則見報錯三及解決方法

HITUAVyolo.yaml數(shù)據(jù)集配置

修改train.py

搜索--data 將coco.yaml改為你自己配置文件的文件名

報錯1

Traceback (mostrecent call last):

File"D:\anaconda3\envs\yolov5\lib\site-packages\git\__init__.py", line89, in <module>

refresh()

File"D:\anaconda3\envs\yolov5\lib\site-packages\git\__init__.py", line76, in refresh

if not Git.refresh(path=path):

File"D:\anaconda3\envs\yolov5\lib\site-packages\git\cmd.py", line 392, inrefresh

raise ImportError(err)

ImportError:Bad git executable.

The gitexecutable must be specified in one of the following ways:

- be included in your $PATH

- be set via $GIT_PYTHON_GIT_EXECUTABLE

- explicitly set via git.refresh()

All gitcommands will error until this is rectified.

This initialwarning can be silenced or aggravated in the future by setting the

$GIT_PYTHON_REFRESHenvironment variable. Use one of the following values:

- quiet|q|silence|s|none|n|0: for nowarning or exception

- warn|w|warning|1: for a printed warning

-error|e|raise|r|2: for a raised exception

解決方法

import git 或 from git import xxx前面加一行

os.environ["GIT_PYTHON_REFRESH"]= "quiet"

參考入門Anaconda cuda yolov5 安裝排坑_anaconda yolov5_Theory D的博客-CSDN博客

報錯2（可能是因為打開了todesk）

解決方法

關(guān)閉todesk,修改model/yolov5s.yaml的種類為自己數(shù)據(jù)集的class數(shù)目 nc

報錯3

AssertionError:train: No labels found in D:\code\datasets\HIT-UAV\train.cache, can not starttraining. See https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data

問題解決

參考一步真實解決AssertionError: train: No labels in /xxx/xxx/xxx/datasets/VOC_To_YOLO/train.cache._assertionerror怎么解決_藍胖胖?的博客-CSDN博客

這是因為【datasets.py】中的圖片路徑設(shè)置的不對

解決方案如下：

1.打開在utils/dataloaders.py

2.打開之后，按【Ctrl】+【F】打開搜索功能，輸入【Define label 】找到下面的一行內(nèi)容：

sa, sb =f'{os.sep}images{os.sep}',f'{os.sep}labels{os.sep}' # /images/,/labels/ substrings

3.由于我數(shù)據(jù)集圖片放在了JPEGImages文件夾內(nèi)：需要把【JPEGImages】替換掉images

再次運行【trian.py】

python train.py --img 640 --batch 16 --epochs 100 --data data/HIT-UAVyolo.yaml--weights yolov5s.pt --device 0

成功

關(guān)于Linux上yolov3/v5環(huán)境配置及訓(xùn)練過程，可以參考：

（linux）yolov5/v3-pytorch目標檢測1：環(huán)境配置gpu+anaconda+pycham+ RTX2080ti 筆記_chao_xy的博客-CSDN博客

（linux）yolov5/v3-pytorch目標檢測2：訓(xùn)練自己的數(shù)據(jù)集，gpu，pycharm, RTX2080ti，單/多顯卡訓(xùn)練，筆記_chao_xy的博客-CSDN博客文章來源地址http://www.zghlxwxcb.cn/news/detail-421435.html