yolov4 darknet模型转OPENVINO的IR模型reshape出错

ERROR:cannot reshape array of size 181944 into shape(1,22743,4)

出错的原理看这里:https://blog.csdn.net/Netceor/article/details/107330685

但是我实际是转模型的,只能看到这里的报错,print的相关信息已在tu中代码里注释了:在这里插入图片描述
此时报错是第三步骤:

#linux  default OpenVINO path
# 1. weights转换成pb模型
python convert_weights_pb.py --class_names cfg/my_names.names --weights_file yolov4_my_versions.weights --data_format NHWC -size 608

# 2. 进入mo.py所在文件夹 转换pb模型到IR模型
python mo.py --input_model frozen_darknet_yolov4_model.pb --transformations_config yolov4.json --batch 1 --reverse_input_channels

# 3. 用IR模型进行推理
python object_detection_demo_yolov3_async.py -i cam -m frozen_darknet_yolov4_model.xml  -d CPU

调试进行了两步检查:

  1. 在第一步的时候,模型是我自己准备的,训练的大小是608*608,不加size是默认416的
  2. 重点还是检查第二部的transformations_config yolov4.json,这个json是根据cfg文件修改的,其中各个参数的定义如下:
	id 和 match_kind 是参数,此处不需要进行修改;

		custom_attributes 主要存储了YOLOv3某些特殊的属性:

      	classses, coords, num, masks,  这些按照Darknet的.cfg配置过来就好;

       anchors:  是一个非必要参数,在模型推理时不使用,主要在解析Region 层的输出;

       entry_points 是一个节点名list,此处不需要修改;
————————————————
版权声明:本文为CSDN博主「caicaiatnbu」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/caicaiatnbu/article/details/103806208

检查后发现自己classses, coords, num, masks, 这些按照Darknet的.cfg配置的,但是id改了一下,原来是这样的:


[
  {
    "id": "TFYOLOV3",
    "match_kind": "general",
    "custom_attributes": {
      "classes": 80,
      "anchors": [10, 13, 16, 30, 33, 23, 30, 61, 62, 45, 59, 119, 116, 90, 156, 198, 373, 326],
      "coords": 4,
      "num": 9,
      "masks":[[6, 7, 8], [3, 4, 5], [0, 1, 2]],
      "entry_points": ["detector/yolo-v3/Reshape", "detector/yolo-v3/Reshape_4", "detector/yolo-v3/Reshape_8"]
    }
  }

为了区别yolov3我改成了yolov4,但是仔细看了一下github上作者的FAQ,发现不能修改id,即使是yolov4,也要写成yolov3的id。
在这里插入图片描述
最后分析了一下,这里的181944 = 22747 x 8,所以不能reshape成181944 = 1 x 22747 x 4。
这种错误我也找了很多相关资料,有人分析是这样的,我觉得很有道理,但是最后没能自己找到深层原因。
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
YOLOv3 Darknet模型训练后不出框框可能是由于以下几个原因造成的。 1. 数据集问题:数据集可能存在标注错误或缺失标注信息的情况。因为YOLOv3是基于边界框的目标检测算法,如果数据集中目标未正确标注或者标注位置不准确,模型就无法准确地检测目标并生成边界框。 2. 参数设置问题:训练过程中使用的参数设置可能不正确,比如学习率设置过高或过低,训练批次数不足等。这些参数设置不合理可能导致模型无法充分学习目标检测的特征。 3. 模型架构问题:YOLOv3的网络结构比较复杂,包含多个层级的特征提取和预测,如果模型的层级结构不正确或者网络层数不够深,就可能导致模型无法准确地检测目标并生成边界框。 解决这个问题可以按照以下方法进行尝试: 1. 仔细检查数据集是否存在标注错误或缺失标注信息的情况,确保数据集中的目标都正确标注。 2. 调整训练参数,根据实际情况适当调整学习率、训练批次数等参数,可以尝试在较小学习率下进行长时间训练,观察是否能够得到更好的结果。 3. 检查模型的网络结构是否正确,是否包含了YOLOv3所需的多个层级的特征提取和预测部分。可以参考YOLOv3模型的官方实现或其他已经经过验证的实现进行对比。 4. 通过查看训练过程的日志输出和模型的评估结果,进一步分析问题所在,可能有助于定位和解决问题。 最后,要注意的是YOLOv3 Darknet模型训练是一个复杂的过程,需要耐心和实践来优化模型并获得准确的目标检测结果。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值