Toybrick

标题: tensorflow 1.14 yolov3 load 模型成功，但提示不支持shape>4？ [打印本页]
作者: mittens    时间: 2020-10-23 18:39
标题: tensorflow 1.14 yolov3 load 模型成功，但提示不支持shape>4？
本帖最后由 mittens 于 2020-10-23 18:41 编辑


复制代码
  1.     # # Create RKNN object
  2.     rknn = RKNN()

  4.     # pre-process config
  5.     print('--> config model')
  6.     rknn.config(channel_mean_value='103.94 116.78 123.68',
  7.                 reorder_channel='0 1 2')

  9.     # Load tensorflow model
  10.     print('--> Loading model')
  11.     ret = rknn.load_tensorflow(
  12.                         tf_pb='wzsb-v50699_601.pb',
  13.                         inputs=['input/input_data'],
  14.                         outputs=['pred_sbbox/concat_2', 'pred_mbbox/concat_2', 'pred_lbbox/concat_2'],
  15.                         input_size_list=[[416, 416, 3]])
复制代码
tensorflow 1.14 yolov3 load 模型成功，但提示不支持shape>4？
有大神知道是什么原因导致，如何调试？


[attach]1383[/attach]
[attach]1384[/attach]



作者: tf小迷弟    时间: 2020-12-28 20:45
我也只这样， 导出来的模型能用吗
作者: jefferyzhang    时间: 2020-12-29 09:05
可以进置顶帖官方rknn qq群里讨论下。
我之前问过他们，他们好像是说这个warning不要紧。
关键是你模型能推理正确就可以了



欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3