Skip to content

A plugin for Mirai that uses Yolov5 for object detection

Notifications You must be signed in to change notification settings

xtaw/ObjectDetectionBot

Repository files navigation

ObjectDetectionBot

初始化工作

Release 下载 coco.names 和 模型 (.torchscript) 并放在 /data/xtaw.objectdetectionbot/yolov5/ 目录下
将模型改名为 model.torchscript
模型共有五种,根据需要自行选择即可。文件越大,准确度越高,速度越慢

使用方法

在群中发送 @Bot + 待检测图片
Bot 便会自动完成检测,并将结果发送在群中
可在一条信息中发送多张图片

配置文件

文件名 /config/xtaw.objectdetectionbot/ObjectDetection.json
第一次启动时会自动生成

示例:

{  
  "Bots": [  
    123456,  
    654321  
  ],  
  "Groups": [  
    123456789,  
    987654321  
  ],  
  "Format": "%AT%\n%IMAGE%\nyolo v5 测试成功 请不要回复\n 用时: %USED_TIME%ms",  
  "UseGpu": false  
}

Bots 中填写 机器人登录的qq号,包含在其中的qq号才会执行目标检测
Groups 中填写 你想执行目标检测的群号,机器人只有在包含在其中的群中被@时才会执行目标检测
Format 中填写 机器人回复格式,一般默认即可,\n 代表 换行%AT% 代表 @执行者%IMAGE% 代表 检测后返回的图片%USED_TIME 代表 检测耗时,单位为 ms
UseGpu 中填写 true或false,若没有gpu则一定填写 false。默认建议 启用gpu

命令

/objectdetection usegpu <state> # 设置是否使用 Gpu, <state> 填写 true / false
/objectdetection bot <add/remove> <id> # 添加和删除 Bots
/objectdetection group <add/remove> <id> # 添加和删除 Groups
/objectdetection format <format> # 设置 Format
/objectdetection list # 显示 Bots 和 Groups 列表
/objectdetection reload # 重新加载配置

/objectdetection 可缩写为 /od
add 可缩写为 a
remove 可缩写为 r

About

A plugin for Mirai that uses Yolov5 for object detection

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages