Ym-Shan / Display-program-for-classification-and-object-detection-algorithms-based-on-PyQt.

A front-end interface for image classification and object detection algorithms built with PyQt5, theoretically capable of showcasing any model or task with minor modifications.

Display-program-for-classification-and-object-detection-algorithms-based-on-PyQt.

A front-end interface for image classification and object detection algorithms built with PyQt5, theoretically capable of showcasing any model or task with minor modifications. The current system includes a classification algorithm designed using the SpikingJelly framework for spiking neural network models, and an object detection algorithm trained on the NEU hot-rolled steel strip metal defect detection dataset using YOLO v8. Additionally, the system features automatic visualization analysis for defect locations, types, and image categories, with demonstrations available in the video.

一个基于PyQt5搭建的图像分类算法和目标检测算法的前端界面，理论上任何模型、任何任务都可以在稍加修改后使用本系统进行展示。本系统目前搭载的分类算法是使用SpikingJelly框架设计的脉冲神经网络模型；目标检测算法是基于YOLO v8的在NEU热轧钢带金属缺陷检测数据集上进行训练的。本系统还包含缺陷位置、缺陷种类、图像类别等数据的自动可视化分析功能，具体演示可见视频。

The system effect is shown in the video

Classification

classification.mp4

Detection

detect.mp4

How to run

1. Configure the environment according to the requirements.txt and pay attention to issues with the opencv and pyqt versions。
2. Run the start_controller.py.

Get full program

Due to size limitations, only the AI related parts have been uploaded for this project. The complete project files can be downloaded from the link below：

Link：https://pan.baidu.com/s/15DqALICN3zOMDk-bGBsc4A?pwd=zs8u 
Password：zs8u