完善资料让更多小伙伴认识你,还能领取20积分哦, 立即完善>
电子发烧友论坛|
我目前使用的转模型代码如下
from ultralytics import YOLOimport cv2import timeimport nncaseimport# 加载预训练的YOLO模型model = YOLO(modelname + ".pt")path = model.export(format="onnx")# 加载ONNX模型onnx_model = onnx.load(modelname + ".onnx")# 设置编译选项compile_options = nncase.CompileOptions()compile_options.target = 'k230'compile_options.preprocess = True # 启用预处理compile_options.swapRB = False # 不交换RB通道compile_options.input_shape = [1, 3, 640, 640] # 设置输入形状compile_options.input_type = 'uint8' # 输入图像数据类型compile_options.input_range = [0, 1] # 输入图像反量化范围compile_options.mean = [0, 0, 0] # 预处理均值compile_options.std = [1, 1, 1] # 标准差设为1,不进行归一化compile_options.input_layout = "NCHW" # 输入数据格式# 创建编译器实例compiler = nncase.Compiler(compile_options)import_options = nncase.ImportOptions()# 导入ONNX模型compiler.import_onnx(onnx_model.SerializeToString(),import_options)# 编译模型compiler.compile()# 生成KModel文件kmodel_data = compiler.gencode_tobytes()with open(modelname + ".kmodel", "wb") as f: f.write(kmodel_data)代码运行无异常,导出模型大小只有12MB,但在设备中运行模型时间非常长(具体超过5分钟一次),而后将预处理关闭,则设备中跑模型会报错 请问,以上问题是哪里的原因?应该怎么解决? 模型速度慢是因为输入分辨率640 * 640太大了,只能用320 * 320的输入分辨率 |
|
相关推荐
|
|
只有小组成员才能发言,加入小组>>
AI_cube训练模型得到了npy文件,没有kmodel文件
770 浏览 0 评论
请问K230 V3.0版本烧录固件和使用IDE到底是烧录哪个啊?
735 浏览 0 评论
443浏览 4评论
AI Cube进行yolov8n模型训练,创建项目目标检测时显示数据集目录下存在除标注和图片外的其他目录如何处理?
462浏览 3评论
autodl算力云在.pt转换onnx文件时正常,但onnx转.kmodel文件时报错,为什么?
552浏览 2评论
如何在大核rtt上把kd_mpi_vicap_start_stream三个摄像头各自出的流拼成一个流呢?
164浏览 2评论
564浏览 2评论
/9
小黑屋| 手机版| Archiver| 电子发烧友 ( 湘ICP备2023018690号 )
GMT+8, 2025-12-2 05:26 , Processed in 0.606257 second(s), Total 73, Slave 55 queries .
Powered by 电子发烧友网
© 2015 bbs.elecfans.com
关注我们的微信
下载发烧友APP
电子发烧友观察
版权所有 © 湖南华秋数字科技有限公司
电子发烧友 (电路图) 湘公网安备 43011202000918 号 电信与信息服务业务经营许可证:合字B2-20210191

淘帖
392
