145基于python目标检测的漂浮垃圾检测

本期给大家介绍的是145基于python目标检测的漂浮垃圾检测。效果图如下：

代码仓库和视频演示地址：

145基于python目标检测的漂浮垃圾检测_哔哩哔哩_bilibili

代码是十分的简介，主要是4个py文件

运行01数据集制作.py会将datasets文件夹下的JPEGImages（jpg或png图片）和Annotations（xml标签）进行读取生成txt文本,里面存放了图片路径和对应的标签。

运行02train.py能够读取txt文本内容结合图片进行训练模型，训练好的模型保存在weights文件夹下。

运行03predict.py可以对单张图片进行检测。

运行04pyqt.py会有个pyqt的可视化界面，通过点击按钮可以加载感兴趣的图片进行识别。