|
本帖最后由 mittens 于 2020-10-23 18:41 编辑
- # # Create RKNN object
- rknn = RKNN()
- # pre-process config
- print('--> config model')
- rknn.config(channel_mean_value='103.94 116.78 123.68',
- reorder_channel='0 1 2')
- # Load tensorflow model
- print('--> Loading model')
- ret = rknn.load_tensorflow(
- tf_pb='wzsb-v50699_601.pb',
- inputs=['input/input_data'],
- outputs=['pred_sbbox/concat_2', 'pred_mbbox/concat_2', 'pred_lbbox/concat_2'],
- input_size_list=[[416, 416, 3]])
tensorflow 1.14 yolov3 load 模型成功,但提示不支持shape>4?
有大神知道是什么原因导致,如何调试?
|
本帖子中包含更多资源
您需要 登录 才可以下载或查看,没有帐号?立即注册
x
|