yolov5模型问题解决SPPF找不到、yolov5smlx.pt文件下载、coco128文件下载、coco128yaml文件更改、coco128数据集not exist、BrokenPipeErro

问题一： 运行detect.py 报错：AttributeError: Can’t get attribute ‘SPPF’ on <module ‘models.common’

原因：models文件夹的comon.py缺少SPPF类，这个类在yolov5的v6.0版本里有，这要复制过来即可，这里直接把代码发出来，省着各位朋友再去找了。

import warningsclass SPPF(nn.Module):# Spatial Pyramid Pooling - Fast (SPPF) layer for YOLOv5 by Glenn Jocherdef __init__(self, c1, c2, k=5):  # equivalent to SPP(k=(5, 9, 13))super().__init__()c_ = c1 // 2  # hidden channelsself.cv1 = Conv(c1, c_, 1, 1)self.cv2 = Conv(c_ * 4, c2, 1, 1)self.m = nn.MaxPool2d(kernel_size=k, stride=1, padding=k // 2)def forward(self, x):x = self.cv1(x)with warnings.catch_warnings():warnings.simplefilter('ignore')  # suppress torch 1.9.0 max_pool2d() warningy1 = self.m(x)y2 = self.m(y1)return self.cv2(torch.cat([x, y1, y2, self.m(y2)], 1))

注意：
（1）如果这个如果这个yolov5s.pt下载不下来的话，可以直接去我的资源里下载，里面包含了（yolov5s.pt、yolov5m.pt、yolov5l.pt、yolov5x.pt、yolov5n.pt）,下载后只要复制到yolo文件夹中就可以使用。

（2）yolov5所在文件夹不要包含中文路径！很重要的一点。

运行detect.py在对应的runs/expn的文件夹里有以下的结果代表detect.py运行成功，yolov5自带两张图片用于检验模型。

问题二： coco128文件下载、coco128yaml文件更改、coco128数据集not exist问题

coco128.zip放在我的资源里了，有需要的可以下载。
1）运行train.py，遇到以下报错：

解决方法：将原文件中的main函数部分：

更改为：

if __name__ == '__main__':parser = argparse.ArgumentParser()parser.add_argument('--weights', type=str, default='', help='initial weights path')parser.add_argument('--cfg', type=str, default='yolov5s.yaml', help='model.yaml path')

2）关于coco128\images\train2017 does not exist 问题

此问题需要对coco128yaml文件更改

这个搞了好久才终于解决，解决办法就是：
在yolov5文件夹中新建一个文件夹，我取名是MyData（任何名字都可以，不要是中文）用于存放coco128的images和labels
将coco128中的images和labels分别复制过来

如下图所示，然后将data文件夹中的coco128.yaml train，val路径更改，即可解决

train: MyData/images/train2017  # 128 images
val: MyData/images/train2017  # 128 images

运行train.py

3) 出现BrokenPipeError，这个是线程问题

将workers的默认参数更改一下就可以了。

yolov5模型问题解决SPPF找不到、yolov5smlx.pt文件下载、coco128文件下载、coco128yaml文件更改、coco128数据集not exist、BrokenPipeErro相关推荐

yolov5模型训练
本文将介绍yolov5从环境搭建到模型训练的整个过程.最后训练识别哆啦A梦的模型. 1.anconda环境搭建 2.yolov5下载 3.素材整理 4.模型训练 5.效果预测 - Anconda环境搭 ...
tensorrt部署YOLOv5模型记录【附代码，支持视频检测】
训练出来的模型最终都需要进行工业部署,现今部署方案有很多,tensorflow和pytorch官方也都有发布,比如现在pytorch有自己的Libtorch进行部署[可以看我另一篇文章有讲利用Libt ...
使用c++onnxruntime部署yolov5模型并使用CUDA加速（超详细）
文章目录前言 1.Yolo简介 2.onnxruntime简介 3.Yolov5模型训练及转换 4.利用cmake向C++部署该onnx模型总结前言接到一个项目,需要用c++和单片机通信,还要 ...
Opencv、dnn部署自己的Yolov5模型记录
Opencv.dnn部署自己的Yolov5模型记录一.环境配置 1.opencv == 4.5.1+dnn模块 2.pytorch == 1.8 3.ubuntu18.04 二.代码来源 1.htt ...
教程：超详细从零开始yolov5模型训练
本文将介绍yolov5从环境搭建到模型训练的整个过程.最后训练识别哆啦A梦的模型. 1.anconda环境搭建 2.yolov5下载 3.素材整理 4.模型训练 5.效果预测 - Anconda环境搭 ...
基于YOLOv5模型压缩、模型量化、模型剪枝
基于YOLOv5模型压缩.模型量化.模型剪枝代码下载地址:下载地址 Requirements pip install -r requirements.txt Pruning for YOLOs Mo ...
使用YOLOv5模型进行目标检测！
↑↑↑关注后"星标"Datawhale 每日干货 & 每月组队学习,不错过 Datawhale干货作者:陈信达,华北电力大学,Datawhale成员目标检测是计算机视觉 ...
(四)训练用于口罩检测的YOLOv5模型
目录介绍准备训练和验证数据在Colab Notebook上训练YOLOv5模型在Google Colab上测试模型下一步在这里,我们将训练和测试用于口罩检测的YOLOv5模型. 介绍在本 ...
IDEA中项目引入独立包打包失败问题解决（找不到包）
IDEA中项目引入独立包打包失败问题解决(找不到包) 参考文章: (1)IDEA中项目引入独立包打包失败问题解决(找不到包) (2)https://www.cnblogs.com/coderdxj/p ...

yolov5模型问题解决SPPF找不到、yolov5smlx.pt文件下载、coco128文件下载、coco128yaml文件更改、coco128数据集not exist、BrokenPipeErro

yolov5模型问题解决SPPF找不到、yolov5smlx.pt文件下载、coco128文件下载、coco128yaml文件更改、coco128数据集not exist、BrokenPipeErro相关推荐

最新文章

热门文章