首先是 github
windows store 直接搜索 watt toolkit 可以加速 github
建议一直开着,不论是在 github 上扒代码还是后续调试,因为后面项目中下载一些东西或许还是会访问到 github 去下载
关于项目需要的库
GitHub - ultralytics/yolov5 at v7.0 ,这是我使用的版本
项目中有个文件 requirements.txt ,里面是作者列出的库清单,且付有一条命令
pip install -r requirements.txt
这里依然来到 Anaconda 命令行,然后进入你编译这个项目时选择的编译环境
conda activate pytorch
把最后的文件名带上在磁盘中的绝对路径(取决于你把项目放在哪,下面的代码是我的位置)
pip install -r D:\yolov7\yolov5_7.0\requirements.txt
当然,还有其他方法,如果你的 IDE 很聪明,或许有一些功能辅助你补全需要的库
也可以不断地运行,缺什么库就去命令行 pip 什么库~
labelImg
一开始我在 Anaconda 中创建的 pytorch 环境下安装
pip install labelimg
安装是成功了,但是一划框就闪退
解决方法:
就是 labelImg 还不支持高版本的 python ,当前所在的名为 pytorch 的环境我用的是 python3.11
创建新环境
conda create -n forlabel python=3.9
进入新环境
conda activate forlabel
在新环境里就可以用辣~
(上述命令都是在 Anaconda Prompt 里完成的哦)
至于如何使用 labelImg
具体参考
【一看就懂,一学就会】yolov5保姆级打标签、训练+识别教程_哔哩哔哩_bilibili
数据集放的位置
注意和项目同级哦
coco128 是程序自行下载的
必须叫 images 和 labels 哦(注意加 s )
关于修改 yaml 文件、进行训练、进行检测还是参考那个视频,该博客记录遇到的没讲到的问题
【一看就懂,一学就会】yolov5保姆级打标签、训练+识别教程_哔哩哔哩_bilibili
开始训练遇到的问题
有报错
【OSError: [WinError 1455] 页面文件太小,无法完成操作】、【OSError: [WinError 1114] 动态链接库(DLL)初始化例程失败】
解决方案来自教程
或许是 GPU 比较一般,又或许是 C 盘空间太少,还出现了闪退的情况
总而言之,在 train.py 中将超参数 --batch_size 从16改为1,问题解决,可以成功训练。