Coder Social home page Coder Social logo

nonebot_plugin_imagelabels's Introduction

nonebot_plugin_Imagelabels

基于yolov5实现的图像标注插件

安装方法

安装: 使用pip :

pip install nonebot_plugin_Imagelabels

在项目内导入:

nonebot.load_plugin('nonebot_plugin_Imagelabels')

或者使用nb:

nb plugin install nonebot-plugin-Imagelabels

使用渲染视频时请将ffmpeg配置于环境变量,否则渲染好也将无法发送

如果想使用GPU渲染请安装好Cudn与Cudnn,方法自行百度

在插件被成功导入到nonebot后将自动检查yolov5-master是否位于Bot根目录,如果没有,将自动下载yolov5-master文件夹于根目录并安装相关依赖

(请注意:执行上述操作时Bot处于堵塞状态,将无法接收并处理其他消息!!这种状态可能持续10多分钟 安装完成后将自动生成parameter.json于根目录 可以自己将yolov5文件夹放置于Bot根目录,并安装相关依赖)

效果及使用

图片标注+图片 如:

图片标注training

image

等待服务端处理完成即可返回处理后的图像

标注视频:

(采用多线程,从而不影响bot正常运行) 在发送标注视频后按照bot提示将视频发送后回复任意消息

如:

image

(注意:处理图像期间Bot也为堵塞状态,无法处理其他消息,过程大约需要5-20s ,默认为CPU渲染,如果想更换GPU,自行百度)

其他: 首次使用需要下载yolov5s.pt / 其他 如果没有,yolov5将从github下下载,国内下载速度极慢,且可能被墙

可以自己用迅雷等其他工具下载pt文件,放置于yolov5-master目录下

下一阶段目标

1.修复更多小问题

2.更改更多细节,优化插件

3.将视频改为自动发出,而不是手动更改

特别感谢

更新日志

v 1.0.0 添加视频标注功能,修复某些小问题

v 0.1.6 :修复了一些可能遇到的问题,并添加了恢复默认功能,在代码出现非网络导致的下载问题时可尝试使用指令"恢复默认"来重置默认值从而解决bug

v 0.1.5 及以下 :可均视为0.1.5版本,没有更新任何东西,只是增加了依赖

关于错误代码

error code 001 :可能由于没有管理员/root权限导致无法重置配件文件导致的 error code 002 :可能由于网络环境较差无法下载yolov5所需文件导致

详细错误问题请参见Bot目录下的Imagelabels-err.txt

其他

本人代码写的可能不好,并且没有注释,可能无法利于代码的阅读,多处变量也重复定义,但请多多指教

若有代码上的问题或意见,欢迎投稿到[email protected],我将一一阅读并恢复

nonebot_plugin_imagelabels's People

Contributors

heartestrella avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.