齐天小圣66 2023-05-27 20:59 采纳率: 50%
浏览 54
已结题

yolov5转onnx验证出错

改过结构的yolov5转onnx,转换成功,但是用detect.py --weights best.onnx验证的时候出现下面报错:
IndexError: The shape of the mask [3, 40, 2] at index 2 does not match the shape of the indexed tensor [3, 40, 6] at index 2
是输出张量不匹配,请问要怎么修改呢?看哪一个文件修改?

  • 写回答

6条回答 默认 最新

  • 喝茶品人生 2023-05-27 21:08
    关注
    获得2.40元问题酬金

    输入输出张量图像检查过了吗,可以去看看输出节点的定义--在模型的ONNX文件里

    评论 编辑记录

报告相同问题?

问题事件

  • 系统已结题 6月4日
  • 创建了问题 5月27日

悬赏问题

  • ¥20 求帮,直连能连上oracle12,但是thinkphp6就是报错
  • ¥15 paddleocr运行报错
  • ¥15 怎么用 matlab 设计滞后-超前串联校正网络
  • ¥15 MFC引用C#生成的dll,将dll放置到非exe程序目录,如何操作
  • ¥15 C#创建webservice接口,三方通过多次跳转访问本方服务,获取wsdl文档,wsdl中ip地址为局域网内本机地址而非三方直接访问的地址。
  • ¥15 关于#wireshark#的问题:需要安卓app流量数据集要安卓流量做包序列长度的实验,比如某些流量是在看视频还是在发评论
  • ¥15 Smail语句如何使用判断语句跳过验证卡密界面
  • ¥15 关于#wireshark#的问题:并且能够给数据做标注,如这个流量是在看视频或者是在转账
  • ¥15 运筹优化,gurobi,python
  • ¥20 springmvc重定向和返回json