更简单一点!PySimpleGUI 快速创建 YOLO 目标检测的界面

本文中的所用到的代码和文件可以从下列地址获取:


使用 GUI (图形用户界面)的优点是用户应该能够立即使用应用程序,并能快速找到和使用他们想要的功能。在日常生活、工作中使用到的应用程序(浏览器、office、微信等等)就是 GUI 的体现,它使我们的工作变得效率更高、门槛更低。

之前在桌面端只试过使用 Matlab 制作 GUI,具体可以查看这篇文章:Matlab GUI app designer 多人消费付款记账催债系统 。使用 Matlab 的 APP designer 或者 Guide (已过时)可以非常方便的制作出界面,因为只要拖拽即可。但是使用Matlab制作界面对于应用的使用者来说非常不便:它打包后的体积非常大,而且需要安装额外的 Matlab 运行库才能够使用。

Python 作为目前非常“流行”的编程语言,实际上确实非常好用。对于我来说,它能够把日常琐碎重复的工作自动化完成,比如说文件重命名、批量替换等。制作图形界面能够让使用者更加方便的使用这些功能,而不是面对漆黑一片的代码行。

在网上看到这么一句话:PySimpleGUI 是适用于人类使用的 Python GUI。它将 tkinterpyQtWxPython 等 Python 知名的 GUI 库转换为可移植的、用户更友好的界面开发库。当然缺点也是显而易见的,PySimpleGUI 只能制作出极为基础的界面,可能无法实现复杂且好看的界面设计。

创建的步骤

界面部分

firebase_login

下文以最简单的 firebase_login 界面为例:

  • 建议安装 Anaconda 管理库文件、使用 Python。
  • 使用 Anaconda Prompt 安装 PySimpleGUI,如果安装过程中提示需要安装其他的安装包,安装即可 :
  • 引入文件:
  • 创建一个 PySimpleGUI 界面。该界面中有两个需要提交的变量:一个是用户邮箱、还有一个是用户密码,点击 OK 按钮之后即可执行验证 try。该界面中有以下几个元素,一个标题和图片、两个输入框、两个按钮:
    • sg.Text 创建文字,可以设置字体、大小、颜色和位置等。
    • sg.Image 引入图像,若其需要和上述的文字排列在同一行,则需要写在同一数组中。
    • sg.OKsg.Cancel 这是两个固定了文字的按钮,当然也可以自定义创建按钮,当点击事件(event)为 Cancel 或者用户点击关闭时,将退出应用;反之,将执行登录的步骤:

yolo_video_with_webcam

下文为主功能界面的部分主要代码,其功能性的代码详见另一篇文章。

  • sg.In 输入框
  • sg.Slider 滑块,可以定义数值范围
  • sg.Button 和上文中类似,但是新定义了一个按钮;点击之后会执行
  • sg.FileBrowse 浏览文件按钮
  • sg.FolderBrowse 浏览文件夹按钮

后一个界面分为左边栏和右边栏:

  • position_elem = win.FindElement('_POSITION_') :寻找界面中 key 为_POSITION_的元素。
  • position_elem.Update(targetPosition):将上面寻找到的元素更新为一个新的值。
  • event, values = win.Read(timeout=0):获取GUI中所有数据和变量。
  • gui_confidence = values['confidence']:获取GUI中的变量。
  • firebase_login.firebaseUploadData(targetPositionObject,timeRightNow):调用firebase_login文件中的firebaseUploadData的函数。

检测部分

检测使用了 OpenCV-Python结合 YOLOv4-darknet 训练得到的 weight 文件,检测部分完整的代码文件可以查看下列地址:0sheldonhuang0/nac-python-gui (github.com)coco.names内包含了需要检测的目标名称,给每个标签配上不同的颜色,以便区分。

加载 YOLO weight 和 Config 文件:

  • 使用 cv2.dnn.readNetFromDarknet加载网络。
  • YOLO 含有很多的图层,getUnconnectedOutLayersNames() 用于提取输出图层的名称。

读取视频或者摄像头中下一帧的数据:

检测每一帧图片,frame 为视频截帧得到的图片:

初始化边界框、置信度、目标种类的数组,循环提取每个输出层,每一层中可能有多个框。提取当前目标的目标对象的种类和置信度,由于可以自行设置置信度的最低值,舍弃检测结果小于最低值的部分。最后得出方框的坐标和其他必要的信息,最后将显示在图片上。

在下方代码后有一大段是为了便于将检测的数据转为可以上传至 Firebase 的 Json 数据,这里便不再赘述。至此,程序可以检测出所输入的视频或者摄像头录像中每一帧图片中所包含的目标物体和它们的位置,并且使用方框、名称和置信度标注出来。

下一步将跳转至上一节中界面部分中的yolo_video_with_webcam界面,并且可以每隔一段时间将图像和目标数据上传至 Firebase 和本地。

使用说明

可以直接从 Github 上面下载本程序的源码,推荐使用 Anaconda 和 Spyder 进行编辑,并且需要主要安装以下一些包。若仍然缺少包,则仅需下载对应的包即可。

  • OpenCV-Python
  • Pyrebase4
  • PySimpleGUI
  • Numpy
  • imutils

在安装完库之后,运行即可看到目标检测的实况。界面如上文的三张图所示,但可能和最终的界面有所不同。用户在正常情况下将执行 exe 文件启动桌面端软件。

  • 首个页面中上方两个输入框可以进行测试:如果用户有目标视频文件和对应的 weight 文件,那么可以从此处导入(注意导入文件和文件夹的路径是相对路径)。用户可以选择置信度和阈值,用户在点击 OK 之后即可进行预测。
  • 若用户的设备有摄像头,则可以勾选下方的使用摄像头选项,点击 OK 之后即可识别和检测摄像头所捕捉到的画面。
  • 用户在识别页面将只能调整检测的置信度和阈值,其他调整选项将放在在设置页面。上传的间隔时间,在目前版本无法调整,但在考虑范围之内。

若用户有本系统的账号和密码,则可以通过左下角的按钮进行登录。如果登录成功,桌面端将会隔一段时间将用户所检测到的目标数据和图像上传至 Firebase。若用户没有登录, 则无法上传成功且在目前版本可能出错。

发表评论

您的电子邮箱地址不会被公开。 必填项已用*标注