YOLO_NCNN_GPU 分类推理和目标检测推理源码

精国软件

站在巨人的肩膀上。
这个模块和人气猫的差不多，但是我这个模块可以直接调用yolo官方导出的ncnn模型
有问题或者一起研究的可以加qq群：981202350
测试了v8v10和v11  理论上v5也支持
比如下面的这些模型
from ultralytics import YOLO

# Load a model  目标检测
model = YOLO('D:\yolov8\yolov8n.pt')  # load an official model
model.export(format="ncnn", dynamic=False, simplify=True,opset=11)
#运行次代码会在yolo环境目录下生成yolov8n_ncnn_model目录，目录下的ncnn模型直接调用


# pose 姿态检测
model = YOLO('D:\\yolov8\\yolov8n-pose.pt')  # load an official model
model.export(format="ncnn", dynamic=False, simplify=True,opset=11)


# 分类检测               
#model = YOLO('D:\\yolov8\\yolov8n-cls.pt')
model = YOLO('runs/classify/train3/weights/best.pt')
model.export(format="ncnn", dynamic=False, simplify=True,opset=11)
#运行次代码会在yolo环境目录下生成runs/classify/train3/weights/best_ncnn_model目录，目录下的ncnn模型直接调用



用到了https://bbs.125.la/forum.php?mod=viewthread&tid=14835268  大佬的内存加载dll。
附件只有模块源码，模块源码内也有调用示例，完整示例放qq群吧，论坛上传不了那么大。另外，pose姿态检测，或者后面更新obb可能都放qq群。

乱跑的和风

感谢分享，很给力！~

sun888

谢谢分享!!

imoling

说实话，yolo这些例子太多了，但是像行为检测，行为识别，脸部识别等这些太少了，楼主这个不错，通过姿势可以衍生出很多东西。

ypx123

v5测试 闪退

yeshuang1

谢谢分享!!

xkAnd

哈哈，这不就是“站在巨人肩膀上”升级版嘛！直接用官方模型，这效率杠杠的，一起搞搞研究呗。

tonc

[内容由于不合规被停止生成，我们换个话题吧]

TK丶渲染

666666666666

eklove

哇，这不就是个宝藏帖嘛！直接调用 YOLO 官方导出的 NCNN 模型，这效率直接拉满，期待后续交流啊！