bournes / PyQt5-YOLOv5

PyQt5 implementation of YOLOv5 GUI 人脸检测

TIPS

This repo is based on YOLOv5 v6.1

Download the models of YOLOv5 v6.1 from here，and put the them to the pt folder. When the GUI runs, the existing models will be automatically detected.

Other versions: v5.0, ...

Updated Date：2022/12/4

Demo Video：

https://www.bilibili.com/video/BV1sQ4y1C7Vk?spm_id_from=333.999.0.0

Quick Start

conda create -n yolov5_pyqt5 python=3.8
conda activate yolov5_pyqt5
pip install -r requirements.txt
python main.py

Function

1. support image/video/webcam/rtsp as input
2. change model
3. change IoU
4. change confidence
5. set latency
6. paly/pause/stop
7. result statistics
8. save detected image/video automatically

You can find ui files in main_win and dialog