huange888 / yolov5_7.0_pyside6_active_learning

基于yolov5_7.0 pyside6 active_learning 开发的人工智能主动学习外周血细胞目标检测系统
https://github.com/huange888/
GNU General Public License v3.0
11 stars 1 forks source link
active-learning pyqt pyside6 yolov5

这个是我的开源 pyside6 + yolov5+active-learning项目,欢迎 star395494A6.png and follow!

联系方式

qq:3100417459 欢迎交流

项目介绍

我把一些文件给删掉了 例如segement 什么的
可以在原生的yolov5 7.0重新安装
最主要的就是两个文件
分别是ui.py MainQt.py
直接运行MainQt.py即可
最主要的pt文件就是data/models/biology.pt
这是用yolov5框架+主动学习跑出来的外周血细胞的模型
目前精度0.8 精度较高
并且支持切换模型 切换到原生的yolov5m.pt 支持对其他目标进行识别
但是biology.pt是用商业数据跑出来的模型 暂未考虑对外开放
对于一般的课设作业 可以将run label train export 停止进程按钮删掉

运行环境

pip
install - r
requirements.txt

运行方法

python MainQt.py

项目截图

运行介绍

该项目是使用PySide6库开发的一个基于YOLOv5的图像识别应用程序。
用户可以通过该应用程序选择图像文件或摄像头实时检测目标物体,并显示检测结果。
同时,该应用程序还支持主动学习功能,用户可以手动标记一些错误的检测结果,以帮助模型进行优化。
不过目前摄像头还没开发好 只能识别图片 同时对于视频识别可以用原生代码进行检测 在qt中实现运行较慢
暂未考虑实现视频识别模块

项目结构

功能区

启动按钮 (Start Button)