search

eecn / ncnn-android-yolov8-pose

32 stars 5 forks source link

尝试转换yolov8m或者yolov8l转ncnn后做测试 #5

Open jackspeed opened 3 months ago

jackspeed commented 3 months ago

大佬 您好！请问下您有尝试转换yolov8m或者yolov8l转ncnn后做测试吗？我把yolov8l转ncnn后，用您这个项目跑会闪退。是不是我转的步骤有问题？请请教一下您是怎么转的呢。我使用官网的export转的： from ultralytics import YOLO model = YOLO("yolov8n-pose.pt") model.export(format='ncnn', half=False, imgsz=640)

按这个步骤，转换出来的YOLOv8n、YOLOv8s都能成功运行，YOLOv8m/YOLOv8l加载后就闪退。

image
eecn commented 2 months ago

要扩展m或者l首先要在string.xml文件增加相应字段，然后在yolov8ncnn.cpp文件中Java_com_tencent_yolov8ncnn_Yolov8Ncnn_loadModel函数增加其对应的模型初始化实现。