简介
通常我们使用YOLOv5训练完成模型以后就会利用detect.py进行推理,但是非可视化的推理总是体验不好,为此我们开发了一个带界面的YOLOv5推理程序,该程序不仅可以进行图片、视频、摄像头推理,而且更可以拓展出数据库等其他功能。
- 环境:Python3、YOLOv5官方环境、flask
- 硬件需求:Windows(GPU/CPU)、Ubuntu(GPU/CPU)
目录
.
├── data
├── detect.py
├── detect_logical.py
├── hubconf.py
├── lib
├── main_logic.py
├── models
├── output
│ └── img_output
│ ├── 2021-08-16-19-12-13.jpg
│ └── 2022-05-18-12-37-30.jpg
├── requirements.txt
├── test.py
├── train.py
├── tutorial.ipynb
├── ui
│ ├── __pycache__
│ │ ├── detect_ui.cpython-37.pyc
│ │ ├── detect_ui.cpython-39.pyc
│ │ ├── login_ui.cpython-37.pyc
│ │ ├── login_ui.cpython-39.pyc
│ │ ├── registe_ui.cpython-37.pyc
│ │ └── registe_ui.cpython-39.pyc
│ ├── detect_ui.py
│ ├── login_ui.py
│ ├── ori_ui
│ │ ├── detect_ui.ui
│ │ ├── login_ui.ui
│ │ ├── registe_ui.ui
│ │ └── register_ui.ui
│ ├── registe_ui.py
│ └── register_ui.py
├── ui_img
│ ├── icon.jpg
│ ├── logo.jpg
│ └── small_log.png
├── userInfo.csv
├── utils
└── weights
├── download_weights.sh
├── yolov5m.pt
└── yolov5s.pt
特点
- UI界面与逻辑代码分离
- 支持自选定模型
- 同时输出检测结果与相应相关信息
- 支持图片,视频,摄像头检测
- 支持视频暂停与继续检测
使用
- 购买源代码:面包多-【源码】YOLOv5检测GUI界面(带登录)
- 下载或训练一个模型,将“.pt”文件放到weights文件夹
- (运行检测)直接运行detect_logical.py,进入检测界面
- (带登录的运行检测)运行main_logical.py,先登录,在进入检测界面
温馨提示:
1.项目放置在全英文路径下,防止项目报错
2.默认使用cpu进行检测,自己可以在init中手动切换GPU