1、测试项目。最开始运行时,需要使用别人的权重,运行python detect.py文件,测试这个项目是否能够正常运行。 2、更改数据集。将data中数据集进行更改,images中放入自己的数据。labels中放入与数据一一对应的标签,保证数据与标签是一一对应的关系。 3、训练数据。运行python train.py文件,会在weights权重文件夹中,或者是runs文件中的最后一个文件中生成一个weights文件夹,如下图所示。 如果在项目下的weights文件中生成,就不再复制粘贴,如果是在runs文件夹中生成,就复制到项目yolov5-master下的weights文件中,将上图的best.pt覆盖weights中best.pt文件。如下图所示。
4、测试数据。更改inference中的images数据集。如下图所示,output文件夹保持为空即可。这个文件夹主要用来存放自动打标签的测试数据。
通过上图,主要更改上面的路径即可。 1、
2、
3、runs文件中exp文件可以全部删除。跑代码会自动生成。
torch 1.6 torchvision 0.7 如果中间缺少什么包,就补充什么包就行。另外,
pip install -U requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple1、出现keyError错误 参考链接:https://blog.csdn.net/weixin_41010198/article/details/108239897 把images和label目录下的,除了图片数据集和标签文件外的其他文件统统删除。 2、需要Xmanager软件来处理X11转发需求
参考链接:https://www.xshellcn.com/xsh_column/jiaocheng-x11zf.html 下载Xmanager即可解决,甚至不需要上述链接的操作步骤。
3、可能会出现的问题:版本问题
最开始运行的是yolov5 的1.0或者是2.0的版本,使用的框架高于要求的框架,会出现
torch.nn.modules.module.ModuleAttributeError: ‘BatchNorm2d‘ object has no attribute 的错误。 之前还以为是代码有问题,所以真的要多听听他人的建议,不要一意孤行。也是因为代码看不懂,才取消了改代码的想法。hhhh,卑微。 如果出现上述错误,可以参考: https://shliang.blog.csdn.net/article/details/108219810 https://shliang.blog.csdn.net/article/details/108124252