Toybrick

rknn_batch_size问题

lixunhuan

新手上路

积分
19
发表于 2025-7-1 17:28:08    查看: 154|回复: 1 | [复制链接]    打印 | 显示全部楼层
如题所示,当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是否可以提高推理效率呢?
回复

使用道具 举报

lixunhuan

新手上路

积分
19
 楼主| 发表于 5 天前 | 显示全部楼层
自己解决了这个推理问题,但是请问 RKNN的 多batchsize是否可以提高推理效率呢? 我这边没有实际验证过
回复

使用道具 举报

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

本版积分规则

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


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