Toybrick

rknn_batch_size问题

lixunhuan

新手上路

积分
5
楼主
发表于 5 小时前    查看: 14|回复: 0 | [复制链接]    打印 | 只看该作者
如题所示,当ret = rknn.build(do_quantization=do_quant, dataset=DATASET_PATH,rknn_batch_size=8).设定为8时推理部分的代码应该如何去进行改进才能正常进行推理呢?
当前报错:
Traceback (most recent call last):
  File "/home/admin/projects/rknn/rknn_model_zoo/examples/yolov7/python/yolov7.py", line 272, in <module>
    draw(img_p, co_helper.get_real_box(boxes), scores, classes)
  File "/home/admin/projects/rknn/rknn_model_zoo/examples/yolov7/python/yolov7.py", line 167, in draw
    print("%s @ (%d %d %d %d) %.3f" % (CLASSES[cl], top, left, right, bottom, score))
IndexError: tuple index out of range


多batchsize是否可以提高推理效率呢?
回复

使用道具 举报

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

本版积分规则

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


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