search

wkt / YoloMobile

A Android Library for YOLOv5/YOLOv7/YOLOv8 Detection and Pose Inference Based on NCNN
Other
43 stars 11 forks source link

.pt to ncnn ? #2

Closed atengya closed 9 months ago

atengya commented 11 months ago

大佬，我用你的代码跑yolov8的模型，但是无法识别。我是按照网上的转换教程来的，不知道出了什么问题。能不能讲一下如何转换模型和设置.json

atengya commented 11 months ago

OK，成功了，使用YOLOv8，使用yolov8自带的 model.export(format='ncnn') 直接转换成ncnn。 前面我使用的是转换成onnx，再转换成ncnn，不知道为什么会失败

wkt commented 11 months ago

OK，成功了，使用YOLOv8，使用yolov8自带的 model.export(format='ncnn') 直接转换成ncnn。 前面我使用的是转换成onnx，再转换成ncnn，不知道为什么会失败

pt->onnx->ncnn和model.export(format='ncnn')模型输出数据格式不一样 目前只支持model.export(format='ncnn')