之前在官网下载了yolov5的代码之后按照博客里的讲解一步步操作总是不能正常运行,甚至一度把conda中的一些重要安装包给删了。现在已运行成功。
我使用的是ubuntu18.04系统,GPU1080ti,pytorch框架。
- 下载官网的yolov5代码。我比较习惯将其重命名为yolov5.
- 将下载好的权重(yolov5l.pt,yolov5m.pt,yolov5s.pt,yolov5x.pt这四个,我下载的权重是4.0版的)放在工程文件夹内。
- 将下载好的coco128数据集放在与工程文件夹并列的地方而不是文件夹内。
- 运行detect.py.
- 运行train.py.
- 运行test.py.可以看到运行结果保存在runs/test/exp中。