使用 YOLOX 检测图片中的奶龙、动漫风格人物、龙图等目标。

💦 加入QQ群 (978132437)

你说上面那句话有个等字?

黄桃罐头: 你好!

模型

请前往 Releases 中下载模型。labels.txt 为分类标签文件。

评估

结果仅供参考。

模型	尺寸	mAPval 0.5:0.95	AP nailong	评估验证集	镜像地址
nailong_v2.6_tiny	416,416	53.5	73.474	v2.6
nailong_v2.5_tiny	416,416	48.3	75.146	v2.5
nailong_v2.4_m	640,640	48.3	71.763	v2.5
nailong_v2.4_tiny	416,416	48.5	65.682	v2.5
nailong_v2.3_m	640,640	52.0	82.614	v2.4	下载
nailong_v2.3_tiny	416,416	51.7	68.940	v2.3	下载

使用

直接推理

请参考 YOLOX 的 ONNX 推理脚本。

python onnx_inference.py -m "nailong_v2_m.onnx" -i "输入图片" -o "输出目录" -s 0.3 --input_shape 640,640

搭建API服务

nkxingxh/yolox-onnx-api-server

搭建API服务给其他程序调用。

python server.py -m nailong_v2.1_m.onnx -l labels.txt

MiraiEz 图片过滤

MiraiEz 是一个支持 mirai-api-http 的 PHP 机器人框架。 方便、快速、高效地使用 PHP 编写你自己的 Bot。

群图片过滤插件

使用插件前, 需要先搭建API服务。

安装插件后, 在 BOT 所在的任意群发送一张任意图片, 插件会自动生成配置。

修改配置中的 classes 部分即可指定目标类别。具体选项说明请查看生成的配置文件。

"classesDesc": "可以配置多个类别组",
"classes": [
    {
        "classId": [1],
        "className": ["nailong"],
        "score": 0.7,
        "actionDesc": "可以对每一组类别单独设置动作",
        "action": {
            "recall": false,
            "reply": "拒绝唐龙",
        },
    },
    {
        "classId": [],
        "className": ["long"],
        "score": 0.75,
        "action": {
            "recall": false,
            "reply": "龙图也是龙"
        }
    }
]

nonebot-plugin-nailongremove

Refound-445/nonebot-plugin-nailongremove

NailongRemove 是一款奶龙识别插件，可以识别群中的奶龙表情包并撤回该表情。

安装插件后, 修改配置中的 NAILONG_MODEL 为 1 即可。

插件默认将使用最新版本的 tiny 模型。

其他

如果你的项目也使用了本模型, 可以打开 Issue 介绍你的项目并添加在列表中。

声明

NailongDetection 根据 AGPL-3.0 许可证进行许可，有关详细信息，请参阅 LICENSE 文件。

本项目仅用作学习 YOLOX 模型训练与推理，且本项目没有任何承诺与保证。