eecn / ncnn-android-yolov8-pose

34 stars 5 forks source link

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

Closed jackspeed closed 2 days 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 3 months ago

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