Toybrick

楼主: kaishi

FusedBatchNormV3 support

cfan

注册会员

积分
193
楼主
发表于 2020-12-23 11:49:24 | 显示全部楼层
jefferyzhang 发表于 2020-3-24 16:04
Yes, it is a perf without quantization.
You must set do_quantization=True, and run again.

我在你的代码中看到有这么一句话,
# Workaround RKNN 1.1.0 bug
想问问是遇到了什么样的bug。
我遇到了一个sortmax层相关的bug,
其输出的结果在rknn和在直接pycharm调用pb模型上输出的结果有较大差异
回复

使用道具 举报

cfan

注册会员

积分
193
沙发
发表于 2020-12-24 13:53:56 | 显示全部楼层
jefferyzhang 发表于 2020-12-23 11:57
这个bug早在1.2.0就解决了。。。

再问一下,我把模型用一张图片进行量化之后速度达到7帧左右,还有什么办法能提高速度吗
回复

使用道具 举报

cfan

注册会员

积分
193
板凳
发表于 2020-12-28 15:25:07 | 显示全部楼层
jefferyzhang 发表于 2020-12-24 14:13
可以进置顶帖qq群里跟他们讨论下。

问了,没人理,,,我用一张图像量化FPS是6.65,用两百张图像反而成了6.63。用的是python接口,连接1808评价得到的。
  1.     # Config for Model Input PreProcess
  2.     rknn.config(channel_mean_value='127.5 127.5 127.5 127.5', reorder_channel='0 1 2',batch_size = 8)
  3.     #rknn.config(channel_mean_value='0 0 0 1', reorder_channel='0 1 2')

  4.     # Load TensorFlow Model
  5.     print('--> Loading model')
  6.     rknn.load_tensorflow(tf_pb='./model.pb',
  7.                          inputs=['lanenet/input_tensor'],
  8.                          outputs=['lanenet_model/vgg_frontend/vgg16_decode_module/binary_seg_decode/binary_final_logits/binary_final_logits'],
  9.                          input_size_list=[[256, 512, 3]])
  10.     print('done')

  11.     # Build Model
  12.     print('--> Building model')
  13.     rknn.build(do_quantization=True, dataset='./dataset.txt')
  14.     print('done')

  15.     print('--> Begin evaluate model performance')
  16.     perf_results = rknn.eval_perf(inputs=[img])                #评估模型性能。 返回模型在硬件上运行的总时间
  17.     print('done')
复制代码
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表