win11 yolov5训练踩坑,笔记

环境配置见yolov5/yolov3 pytorch环境配置gpu+windows11+anaconda+pycharm+RTX3050 笔记

正确的训练步骤见(windows)yolov5/yolov3训练自己的数据集,win11 gpu,pycharm,RTX3050笔记

数据集文件分布

建议将JPEGImages改为images,否则见报错三及解决方法

HITUAVyolo.yaml数据集配置
修改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)
解决方法

关闭todesk,修改model/yolov5s.yaml的种类为自己数据集的class数目 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】中的图片路径设置的不对

解决方案如下:

1.打开在utils/dataloaders.py

2.打开之后,按【Ctrl】+【F】打开搜索功能,输入【Define label 】找到下面的一行内容:

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

3.由于我数据集图片放在了JPEGImages文件夹内:需要把【JPEGImages】替换掉images

再次运行【trian.py】

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

成功

关于Linux上yolov3/v5环境配置及训练过程,可以参考:

(linux)yolov5/v3-pytorch目标检测1:环境配置gpu+anaconda+pycham+ RTX2080ti 笔记_chao_xy的博客-CSDN博客

(linux)yolov5/v3-pytorch目标检测2:训练自己的数据集,gpu,pycharm, RTX2080ti,单/多显卡训练,笔记_chao_xy的博客-CSDN博客

  • 4
    点赞
  • 34
    收藏
    觉得还不错? 一键收藏
  • 7
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值