Yolov5代码来源:yolov5/utils at master · ultralytics/yolov5 (github.com)
一开始我本来是下载Yolov5-v5.0版本的,因为我想跟着看小土堆的视频,但是因为目前已经更新到v7.0版本了,以致于我运行的时候出现了很多问题。后面我看了博主发各种问题的解决方法,有些问题实在是找不到方法,然后我就去下了7.0的版本,果然问题少了很多,至少可以运行了。
在训练自己的自创的数据集时出现了很多很多问题,如下:
(1)ImportError: Bad git executable.
这个问题解决地莫名其妙,我去网上搜了解决方法,看到有说出现这个问题是因为没有下载Git这个软件或者没有将Git写入Path,所以我去下载了,也将路径写入了Path,但回去Pycharm运行train.py还是有问题,昨天我就放弃了,然后今天重新跑的时候,这个问题消失了,所以可能是下载完、路径改完之后还得电脑重启??
(2)Exception: Dataset not found.
这个问题提示说我的数据集找不到,然后我就尝试着去改路径。
以下是我yolov5文件的目录概览,“Exception: Dataset not found.”这个报错问题是因为数据集的存放位置和mydata.yaml中数据集对应的相对路径不匹配。
/yolo
├── train.py
├── data
│ ├── coco.yaml
│ └── mydata.yaml
└── mydata
├── images
│ ├── train
│ │ ├── img1.jpg
│ │ ├── img2.jpg
│ │ └── ...
│ └── test
│ ├── img3.jpg
│ ├── img4.jpg
│ └── ...
└── labels
├── img1.txt
├── img2.txt
└── ...
mydata.yaml可以类似以下路径设置:
train: ./my_data/images/train # train images (relative to 'path') 118287 images
val: ./my_data/images/train # val images (relative to 'path') 5000 images
nc : 1
# Classes
names:
0: bottle
(3)attributeerror: 'FreeTypeFont' object has no attribute 'getsize'.
这个报错是因为下载的pillow包是大于10.0版本的(大于10.0版本的删除了getsize功能),我将这个版本卸载了,下载了9.5版本后报错就解决了。(以下方式是通过镜像源安装的,会更快速一些)
pip install pillow==9.5 -i https://pypi.tuna.tsinghua.edu.cn/simple
(4)运行detect.py出来的图片结果没有检测框 ,我是用了自制数据集训练出来的best.pt进行检测,因为我是cpu版的pytorch,怕数据太多跑不动我就先训练了10来张图片,效果不好。但是当我换成yolov5s.pt模型时,十来张图片里面有几张是有检测框的。所以我打算要入门服务器了,尝试多些数据看看能不能准确率高一些!!
最后的最后,跑通了一个模型代码真的很高兴!!!论文有望了!!!