ncnn-android-yolov7跑自己模型的注意事项

ncnn-android-yolov7


这是一个示例 ncnn android 项目,它依赖于 ncnn 库和 opencv

https://github.com/Tencent/ncnn

https://github.com/nihui/opencv-mobile

https://github.com/xiang-wuu/ncnn-android-yolov7

如何构建和运行


步骤1

https://github.com/Tencent/ncnn/releases

下载 ncnn-YYYYMMDD-android-vulkan.zip或自己构建 ncnn for android

将 ncnn-YYYYMMDD-android-vulkan.zip 提取到 app/src/main/jni 中,并在 app/src/main/jni/CMakeList 中将ncnn_DIR路径更改为您的路径.txt

步骤2

https://github.com/nihui/opencv-mobile

下载opencv-mobile-XYZ-android.zip

将opencv-mobile-XYZ-android.zip提取到app/src/main/jni中,并在app/src/main/jni/CMakeLists中将OpenCV_DIR路径更改为您的路径.txt

步骤3

使用Android Studio打开这个项目,构建它并享受!

一些注意事项

  1. 安卓 ndk 相机用于最佳效率
  2. 由于缺少HAL3相机接口,崩溃可能发生在非常旧的设备上
  3. 所有模型均手动修改以接受动态输入形状
  4. 大多数小型模型在 GPU 上的运行速度比在 CPU 上的运行速度慢,这很常见
  5. 由于相机曝光时间较长,在黑暗环境中的FPS可能较低

跑自己的模型


流程:pt->onnx->ncnn。

1、yolov7(pt->onnx)

下载代码和权重,这里用的是tiny。

代码用main分支的,权重手动下载release里的。

运行

复制代码
python export.py yolov7-tiny.pt --simplify

运行时要加simplify,不然onnx转ncnn会有不支持的算子。

2、ncnn(onnx->ncnn)

使用下面这个网站转换,要把三个勾都选上。

https://convertmodel.com

3、修改权重输入输出

进入Netron,上传自己的param文件查看权重输入输出(1个输入、3个输出),并更改代码后即可直接编译;

获取晚后在ncnn-android-yolov7/app/src/main/jni/yolo.cpp的Yolo::detect方法修改着几个地方:

复制代码
in0->images
out0->265
out1->279
out2->293

4、修改标注类别名称

不出意外,到这里重新编译一下就可以

由于部分设备不支持camera2,新增JNI方法,传图片进行识别


具体调用看MainActivity处理

THE END


感谢查阅

玉念聿辉:编辑

相关推荐
Uopiasd1234oo11 小时前
上下文引导模块改进YOLOv26局部与全局特征融合能力双重提升
深度学习·yolo·机器学习
饭小猿人11 小时前
Android 腾讯X5WebView如何禁止系统自带剪切板和自定义剪切板视图
android·java
_李小白11 小时前
【android opencv学习笔记】Day 8: remap(像素位置重映射)
android·opencv·学习
美狐美颜SDK开放平台11 小时前
多场景美颜SDK解决方案:直播APP(iOS/安卓)开发接入详解
android·人工智能·ios·音视频·美颜sdk·第三方美颜sdk·短视频美颜sdk
嗷o嗷o11 小时前
Android BLE 里,MTU、分包和长数据发送到底该怎么处理
android
懷淰メ12 小时前
【AI加持】基于PyQt+YOLO+DeepSeek的钢材焊接缺陷检测系统(详细介绍)
yolo·目标检测·计算机视觉·pyqt·缺陷检测·deepseek·钢材缺陷
动物园猫12 小时前
工业织物缺陷目标检测数据集分享(适用于YOLO系列深度学习分类检测任务)
深度学习·yolo·目标检测
Gary Studio13 小时前
Android AIDL HAL工程结构示例
android
迪菲赫尔曼13 小时前
从 0 到 1 打造工业级推理控制台:UltraConsole(Ultralytics + FastAPI + React)开源啦!
前端·yolo·react.js·计算机视觉·开源·fastapi
y = xⁿ14 小时前
MySQL八股知识合集
android·mysql·adb