Skip to content

ncnn安卓部署的yolov5-lite,修改为多个yolov5模型推理

Notifications You must be signed in to change notification settings

Ma0zii/ncnn-android-yolov5-lite

Repository files navigation

ncnn-android-yolov5-lite


因为我需要推理两个不同的yolov5模型

原版的加载推理是同一个

所以在原版的基础上做了些修改

改了后处理部分和一些参数传递

分离了原版jni的部分,写成了cpp来调用推理

因此只需简单修改参数就可以推理多个yolov5模型

使用方法和nihui版的yolov5一样

把ncnn库放进去然后修改cmake即可使用

该版本是推理两个yolov5-lite模型的

CSDN博客:https://blog.csdn.net/Maoziii/article/details/135146324?spm=1001.2014.3001.5502

运行前还需要:

1、准备自己的yolov5模型并转化为ncnn

2、修改target_size,修改模型路径等

3、修改.param文件中permute层的名字与推理时ex获取的out层名字一样

4、修改自己模型.param文件中reshape层,改为动态获取

参考: nihui安卓部署yolov5: https://github.com/nihui/ncnn-android-yolov5

yolov5-lite作者ppogg的仓库: https://github.com/ppogg/YOLOv5-Lite


The YOLOv5 object detection

this is a sample ncnn android project, it depends on ncnn library only

https://github.com/Tencent/ncnn

how to build and run

step1

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

download ncnn-android-vulkan.zip or build ncnn for android yourself

step2

extract ncnn-android-vulkan.zip into app/src/main/jni or change the ncnn_DIR path to yours in app/src/main/jni/CMakeLists.txt

step3

open this project with Android Studio, build it and enjoy!

About

ncnn安卓部署的yolov5-lite,修改为多个yolov5模型推理

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published