This commit is contained in:
haotian 2025-08-19 16:32:01 +08:00
parent af901054af
commit 3ef17b11f1
2 changed files with 5 additions and 3 deletions

View File

@ -9,8 +9,10 @@ rknn = RKNN()
# 配置参数(关键!)
rknn.config(
target_platform="rk3588", # 根据实际芯片型号修改
mean_values=[[123.675, 116.28, 103.53]], # 默认输入是255, 使用这些值归一化到0-1
std_values=[[58.395, 57.12, 57.375]],
# mean_values=[[123.675, 116.28, 103.53]], # 默认输入是255, 使用这些值归一化到0-1
# std_values=[[58.395, 57.12, 57.375]],
mean_values=[[0, 0, 0]], # 默认输入是255, 使用这些值归一化到0-1
std_values=[[255, 255, 255]],
quant_img_RGB2BGR=True,
optimization_level=3, # 最高优化级别
)
@ -27,7 +29,7 @@ ret = rknn.build(
assert ret == 0, "转换 RKNN 失败!"
# 导出 RKNN
ret = rknn.export_rknn("/home/admin-root/haotian/康达瑞贝斯机器狗/rec_shape_20250815.rknn")
ret = rknn.export_rknn("/home/admin-root/haotian/康达瑞贝斯机器狗/rec_shape_20250815_0m.rknn")
assert ret == 0, "导出 RKNN 失败!"

Binary file not shown.