Skip to content

奶龙识别(奶龙检测),基于resnet50,附有训练代码,支持训练所有图片鉴别模型,支持gif和视频识别(检测),这个仓库可能放一些奶龙以外的识别(检测)模型,比如doro识别(doro检测)这种

License

Notifications You must be signed in to change notification settings

spawner1145/NailongRecognize

Repository files navigation

唐死我了,这是一个奶龙鉴别模型及其训练代码和数据集,byd一边做一边笑

基于resnet50训练

release里获取最新模型和数据集

理论上这个库可以进行所有二分类模型训练,只要自己改掉train和test以及相应negative文件夹内的图片再运行train.py就可以做自己想要的图片分类(检测图片中是否有某元素,输出是或否)

支持gif和视频,原理遍历所有帧,有一张有就输出有,否则输出无

我参与开发的机器人已加入奶龙检测功能

二代https://github.com/AOrbitron/Eridanus

一代https://github.com/avilliai/Manyana.git

上面的机器人如果要更换模型打开plugins/nailong11文件夹把里面的nailong.pth替换掉

奶龙识别nonebot机器人插件链接,作者445,https://github.com/superbot-ai445/nonebot-plugin-nailongremove.git

运行 python webui_with_api.py 快速启动推理webui和api服务,会自动读取目录下的所有可选pth模型,但需要额外pip install fastapi uvicorn[standard] gradio

About

奶龙识别(奶龙检测),基于resnet50,附有训练代码,支持训练所有图片鉴别模型,支持gif和视频识别(检测),这个仓库可能放一些奶龙以外的识别(检测)模型,比如doro识别(doro检测)这种

Resources

License

Stars

Watchers

Forks

Packages

No packages published