[RKNN学习记录]RK3588使用ADB进行rknn模型的内存和性能评估_file &quot；rknn api rknn_base

最新推荐文章于 2025-05-02 09:46:05 发布

2401_84008965

最新推荐文章于 2025-05-02 09:46:05 发布

阅读量288

点赞数 2

分类专栏：程序员文章标签：学习 adb 数据库

本文链接：https://blog.youkuaiyun.com/2401_84008965/article/details/138819229

版权

# 创建RKNN
# 如果测试遇到问题，请开启verbose=True，查看调试信息。
#rknn = RKNN(verbose=True)
rknn = RKNN()

# 导入RKNN模型，path参数指定rknn模型路径
print('--> Loading model')
ret = rknn.load_rknn(path=RKNN_MODEL)
if ret != 0:
    print('Load model failed!')
    exit(ret)
print('done')

# 初始化运行时环境，指定连接的板卡NPU平台,device\_id指定前面adb连接的板卡设备ID
# perf\_debug开启进行性能评估时开启debug模式，eval\_mem进入内存评估模式
print('--> Init runtime environment')
ret = rknn.init_runtime(target='rk3588', device_id='10.102.149.127:5555', perf_debug=True, eval_mem=True)
if ret != 0:
    print('Init runtime environment failed!')
    exit(ret)
print('done')

# 模型性能进行评估，默认is\_print是true，打印内存使用情况
print('--> eval\_perf')
rknn.eval_perf()
print('done')

# 调试，模型性能进行评估,默认is\_print是true，打印内存使用情况
print('--> eval\_memory')
rknn.eval_memory()
print('