模型转换
-
【yolov5】pytorch模型导出为onnx模型
博主想拿官网的yolov5训练好pt模型，然后转换成rknn模型，然后在瑞芯微开发板上调用模型检测。但是官网的版本对npu不友好，所以采用改进结构的版本： 将Focus层改成Con…
-
【地平线开发板 模型转换】将pytorch生成的onnx模型转换成.bin模型
1 获取onnx模型 我是使用pytorch训练得到的pth模型，转成onnx模型，这一步之前写过博客，可参考：pytorch分类模型导出onnx模型并验证[0] 2 启动dock…
-
【YOLOV5-5.x 源码解读】export.py
内容 前言 0、导入需要的包和基本配置 1、文件入口 2、parse_opt 3、main 4、run 5、使用 总结 前言 源码：YOLOv5源码.导航:【YOLOV5-5.x …
