yolov7-Pyside6 Image and Video Inference with UI

View Chinese Version here

This project is based on YOLOv7 & PyQt5-YOLOv5

When the GUI is launched, it will automatically detect the presence of the YOLOv7 models that were previously added to the "ptmodel" folder.

With this project, our aim was to offer end users quick inferences that would simplify their work. To achieve this, we focused on developing a lightweight and fast project that would meet their needs.

The system features:

OS : Windows 11 
CPU : Intel(R) Core(TM) i7-10750H CPU @ 2.60GHz   2.59 GHz
GPU : NVIDIA GeForce GTX 1660Ti 6GB

Inferences on Image and Video

Prepare

    ### create virtual env 
    conda create -n yolov7 python=3.9 
    ### requirements
    pip install -r requirements.txt
    ### Download related version Pytorch 
    pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117

Download Pytorch

Run

  python main.py

Function

1. support image/video as input
2. change model
3. change IoU
4. change confidence
5. paly/pause/stop
6. show result
7. export detected image/video

Demo Video

Check in YOLOv7检测界面-Pyside6-GUI

Reference

• WongKinYiu/yolov7
• PyQt5-YOLOv5
• YOLOv7-Predict-with-UI
• Python Qt 简介

End

Please consider giving me a star if this project is helpful to you. ⭐⭐⭐