YOLOv5 模型剪枝RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 512, 8

最新推荐文章于 2024-05-01 11:09:33 发布
最新推荐文章于 2024-05-01 11:09:33 发布
阅读量6k 收藏 23
点赞数 15
分类专栏： yolo v5 文章标签： 剪枝 深度学习 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_43533724/article/details/121743384
版权

在github上直接下载的YOLOv5模型剪枝的代码，修改成自己的数据集后，运行train.py出现：RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 512, 8, 8] to have 1024 channels, but got 512 channels instead

parser.add_argument('--cfg', type=str, default='models/yolov5s.yaml', help='model.yaml path')

修改if name == ‘main’:里的
cfg的dafault为‘models/yolov5s.yaml’即可

派森歪瑞古德
关注
  • 15
    点赞
  • 踩
  • 23
    收藏
    觉得还不错? 一键收藏
  • 9
    评论
专栏目录
team-runtime-error：这是Project 1，Group 4的初始仓库
02-21
团队运行时错误 这是项目1，第4组的初始回购。
解决api-ms-win-crt-runtime-l1-1-0.dll 丢失问题 绿色版 工具下载
07-27
提示api-ms-win-crt-runtime-l1-1-0.dll 丢失，安装这个小玩意就可以解决了
Given groups=1,weight of size[512,1024,1,1],expected input[8, 320, 20, 20] to have 1024 channels, bu
qq_45825952的博客
11-16 6571
关于通道不匹配问题的解决： RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[8, 320, 20, 20] to have 1024 channels, but got 320 channels instead
yolov5运行问题
HUGO-42078367的博客
12-08 1894
2.RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 512, 8, 8] to have 1024 channels, but got 512 channels instead 问题描述： 运行yolov5下面Tags5的代码出现问题： Traceback (most recent call last): File "/media/ls/娱乐/yolov5/yolov5-5.0/train
【yolov5】RuntimeError: Given groups=1, weight of size [24, 64, 1, 1], expected input[1, 128, 32, 32]
woyuxian的博客
05-25 808
图1中出现的问题是因为图2中红线标的层数没加1，因为加了第一条红线的一层，所以以下都要加1。
YOLOV5使用过程中可能出现的问题及解决方法
sixian666的博客
07-09 6151
我尝试过YOLOV5 v5.0 和 YOLOV5 v6.0，都出现了一些问题（小声嘀咕：我刚开始搜索报错原因得到是github社区上作者的回答，结果把它他提供的解决方法都试了一遍，一点作用没有，最后面向csdn解决了问题）所以在此做一个亲测有用的博客的总结1）AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'(6条消息) YOLOV5 训练好模型测试时出现问题：AttributeError: ‘Upsampl
RuntimeError: Given groups=1, weight of size [512, 80, 7], expected input[1, 329, 80] to have 80 ch
oneself的博客
06-13 3114
完整报错： 解析错误： 无分组卷积，卷积核的输入通道是80，输入通道是512,大小为7（一维卷积）（如果是二维卷积，则此处的size是会有4个数，后两个表示卷积核的大小） 实际输入的张量维度是[1,329,80]， 即feature的channel 是329，但期望的输入feature的channel为80通道我这边的实际错误是读入input的shape是[329,80]，所以仅通过 即转置后便解决错误，但如果直接就是不匹配的，要么选择修改网络层的结果（即输入通道设置），要么使用resha
RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 512, 8, 8] to have
qq_44866009的博客
11-27 1万+
RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 512, 8, 8] to have 1024 channels, but got 512 channels instead 我们可以看到默认的模型配置文件yaml路径没有指定 然后我们指定一个之后 然后就可以正常运行了，若是数据集下载出现如下问题 则可以下载给定的下载路径上的压缩包，并将其解压到当前项目所在路径的同级目录下即可 若是报错
【Bug解决】RuntimeError:Given groups=1,weight of size...expected input...but got 3 channels instead.
热门推荐
学习 & 分享 ~
03-14 9万+
报错信息： 原因： 明显是数据读入的通道数不对，应该是 1 通道，但是这里读入的是 3 通道。但是检查了数据，发现就是一通道的灰度图，没错儿呀。最后发现是 模块 打开图像的数据 问题。检查发现，图像竟然是RGB，但我的训练图像是一通道的灰度图，所以得想办法把 mode 转换一下。 解决方法： 这样子网络再读取图像，就是 啦 ~...
yolov5代码显示通道数错误问题RuntimeError: Given groups=1, weight of size 64
weixin_45459097的博客
09-01 4221
问题： 运行train.py时显示通道数错误问题RuntimeError: Given groups=1, weight of size 64 4 7 7, expected input[1, 5, 206, 206] to have 4 channels, but got 512 channels instead。 解决方法： 修改之前的代码： 修改之后的代码： .........
Yolov5目标检测模型运行遇到的相关问题汇总
baidu_35536188的博客
01-09 8254
yolov5运行常见错误
tt100k数据集跑yolov5s模型时，所遇到的问题记录
weixin_46085845的博客
11-17 4470
yolov5遇到的错误
基于yolov5的模型剪枝项目实战源码
10-13
对yolov5s进行稀疏化训练并剪枝，模型参数下降80%，mAP精度几乎不受影响
yolov8s模型进行剪枝源码
05-13
实现步骤: 1. yolov8s模型预训练 2. 模型稀疏化sparsity 3.剪枝 4. finetune 经过键枝后,finetune60个epoch达到原模型迭代52个epoch的mAP值0.78，模型的大小减少了2/5。
mobile-yolov5-pruning-distillation:mobilev2-yolov5s剪枝，蒸馏，支持ncnn，tensorRT部署。超轻但性能更好！
03-19
移动式yolov5修剪蒸馏mobilev2-yolov5s的通道修剪和蒸馏。超轻但性能更好！ TensorRT版本===> Android版本===> 背景yolov5s在640x640分辨率下的计算量和参数量分别为8.39G和7.07M。在速度上仍然有提升空间，通过替换...
基于剪枝后的yolov5模型进行知识蒸馏训练python源码+项目说明.zip
04-12
基于剪枝后的yolov5模型进行知识蒸馏训练python源码+项目说明.zip 基于剪枝后的yolov5模型进行知识蒸馏训练python源码+项目说明.zip 基于剪枝后的yolov5模型进行知识蒸馏训练python源码+项目说明.zip 基于剪枝后的...
YOLOv8模型剪枝源码
03-27
YOLOv8模型剪枝源码
人工智能-项目实践-知识蒸馏-本项目支持对剪枝后的yolov5模型进行知识蒸馏训练
12-26
本项目实现功能如下： 1.训练自己的数据集 2.对任意卷积层进行剪枝 3.剪枝后的训练 4.剪枝后的模型预测 5.利用知识蒸馏对剪枝后模型进行训练
每日OJ题_DFS爆搜深搜回溯剪枝⑤_力扣37. 解数独
最新发布
参考书籍：《C语言程序设计》《数据结构》、《C++ Primer》、《Effective C++》、《STL源码剖析》、《现代操作系统》、《图解TCP/IP》、《图解HTTP》、《Linux高性能服务器编程》、《剑指Offer》、《算法导论》
05-01 1382
每日OJ题_DFS爆搜深搜回溯剪枝⑤_力扣37. 解数独（为了存储每个位置的元素，我们需要定义⼀个⼆维数组。首先记录所有已知的数据，然后遍历所有需要处理的位置，并遍历数字 1~9。对于每个位置，检查该数字是否可以存放在该位置，同时检查行、列和九宫格是否唯一。根据力扣36. 有效的数独的解法，可以使用一个二维数组来记录每个数字在每一行中是否出现，一个二维数组来记录每个数字在每一列中是否出现。）
yolov8剪枝出现如下错误：RuntimeError: Given groups=1, weight of size [64, 64, 3, 3], expected input[16, 32, 60, 84] to have 64 channels, but got 32 channels instead，怎么解决
07-14
这个错误通常是由于模型的输入通道数与预训练模型参数的通道数不匹配导致的。在剪枝过程中，模型的通道数被减少，但是预训练模型参数的通道数仍然保持不变。 要解决这个问题，你可以尝试以下几种方法： 1. 确保剪枝操作与模型的结构相匹配。确保剪枝操作只应用于卷积层的通道维度，而不是其他维度。 2. 检查剪枝操作后的模型结构是否正确。确保剪枝操作后每个卷积层的输入通道数与剪枝后的通道数相匹配。 3. 如果你是基于开源项目实现的剪枝算法，尝试查看相关文档或提问社区，以了解是否有其他人遇到过类似的问题，并找到解决方案。 4. 如果你自己实现了剪枝算法，可以检查你的实现是否有错误，并确保剪枝操作被正确应用。 希望以上方法可以帮助你解决问题！如果仍然有困惑，请提供更多关于剪枝过程和代码实现的细节，我会尽力提供更具体的帮助。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 9
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值