- 如何 onnx 模型可视化。
- pytorch 到 ncnn,有些网络层没有对应的 c++ 底层实现,怎么自己 手撸 ncnn ,或者 修改 .param 文件。
- 认识无法转化的 crop 层。
- 直接下载 yolov5 和 安卓文件,进入到 ncnn-android-yolov5/app/src/main/ ,解压 jni.zip。
-
在 yolov5/models/pts/ 下提供了两个官方的预训练模型。
-
参照 yolov5 官方教程即可,在训练结果中得到 best.pt 文件。
-
使用 export.py 导出 onnx 格式,注意,必须要 带有 --train ,例如
python export.py --train
-
使用 模型剪枝工具转化 onnx 文件
pip install onnx coremltools onnx-simplifier python -m onnxsim best.onnx best-sim.onnx
-
将 剪枝后的文件 best-sim.onnx 转化为 ncnn 文件用于部署,在 这里 转换。
-
将 .param 和 .bin 替换 安卓工程的 ncnn-android-yolov5/app/src/main/assets/ 即可
-
更改 .param 文件,参照 这里 即可