背景:
最近在研究YOLOv8的应用,并且已经在自己的笔记本环境中跑通了YOLOv8的检测和分类算法,训练、验证、预测等功能均已实现;也通过自己的数据集训练出了自己的模型(权重);且之前也做了一个webUI界面,对YOLOv8检测和分类的结果进行展示;但是如果在本地的GUI界面调用摄像头肯定比webUI调用摄像头要方便,因此想在电脑本地做一个GUI界面,对yolov8的分类结果进行展示。
总体逻辑是,打开电脑的摄像头,然后需要分类时,抓拍,对抓拍的当前帧进行分类,并显示分类结果。我在代码里加入了对不同的分类结果进行LED灯显示,如果不需要,不装LED灯,忽略即可;不影响分类功能。
先附上最终的效果图:
程序运行后,会打开一个GUI界面(左图),点击Capture Frame后,对当前帧进行分类,并显示分类结果(右图)。
文章目录
- 一、安装YOLOv8
- 二、搭建GUI
一、安装YOLOv8
既然是对YOLOv8的结果进行可视化