摘要： 水面目标物检测识别是计算机视觉和图像处理技术在特定场景下的应用，主要目的是对水面上的物体进行识别、定位和跟踪。这一技术在水面漂浮物检测、水质监测、航道安全、水上救援等领域具有重要的应用价值。本文基于YOLOv10/v9/v8深度学习框架，通过7647张水面目标物的相关图片，基于yolo框架进行水面目标物的目标检测模型训练，可以检测出14种目标：**["bridge", "ship", "boat", "ball", "rubbish", "rock", "buoy", "platform", "habor", "mast", "tree", "animal", "grass", "person"]** ，同时对相关模型进行了对比，全面对比分析了YOLOv8、YOLOv9、YOLOv10这3种模型在验证集上的评估性能表现。最后基于训练好的模型制作了一款带UI界面的水面目标物检测识别系统，更便于功能演示，保存检测结果。本文提供了完整的Python代码和使用教程，给感兴趣的小伙伴参考学习，完整的代码资源文件可公众号获取。
一、研究背景
水面目标物检测识别技术在水环境治理、航道安全、水上救援等领域的应用前景广阔，可以起到重要作用。运用深度学习技术，YOLOV10\YOLOV9\YOLOV8深度学习框架训练水面目标物检测模型，系统可以高效、准确的识别水面目标物，水面目标物检测识别系统将更加智能化、精准化和高效。水面目标物检测识别系统是一项具有广泛应用前景和重要意义的技术系统，随着技术的不断进步和应用场景的不断拓展，该系统将在未来发挥更加重要的作用。
其主要应用场景包括：
- 环保监测：用于检测水面上的漂浮物（如垃圾、油污等），为环保部门提供数据支持。
- 航运安全：监测航道上的船舶，预防碰撞等安全事故的发生。
- 渔业管理：监测渔网、浮标等渔业设施，提高渔业生产效率和安全性。
- 军事侦察：用于水面目标的侦察与监视，为军事行动提供情报支持。
- 智慧城市：作为智慧城市的一部分，为城市管理者提供水面监控与管理的数据支持。
总的来说，水面目标物检测识别系统的应用不仅提高了效率，提升系统的处理速度，实现更高效的实时检测与识别。结合大数据、人工智能等技术，为决策者提供更全面、更智能的决策支持。随着技术的不断成熟和成本的降低，系统将在更多领域得到应用。
二、主要工作内容
本文的主要内容包括以下几个方面：
-
搜集与整理数据集：搜集整理实际场景中水面目标物的相关数据图片，并进行相应的数据处理，为模型训练提供训练数据集；
-
训练模型： 基于整理的数据集，根据最前沿的YOLOv10/YOLOv9/YOLOv8目标检测技术 训练目标检测模型，实现对需要检测的对象进行实时检测功能；
-
模型性能对比： 对训练出的3种模型在验证集上进行了充分的结果评估和对比分析，主要目的是为了揭示每个模型在关键指标（如Precision、Recall、mAP50和mAP50-95等指标）上的优劣势。这不仅帮助我们在实际应用中选择最适合特定需求的模型，还能够指导后续模型优化和调优工作，以期获得更高的检测准确率和速度。最终，通过这种系统化的对比和分析，我们能更好地理解模型的鲁棒性、泛化能力以及在不同类别上的检测表现，为开发更高效的计算机视觉系统提供坚实的基础。
-
可视化系统制作： 基于训练出的目标检测模型，搭配Pyside6制作的UI界面，用python开发了一款界面简洁的水面目标物检测识别系统，可支持图片、视频以及摄像头检测，同时可以将图片或者视频检测结果进行保存。通过图形用户界面（GUI），用户可以轻松地在图片、视频和摄像头实时检测之间切换，无需掌握复杂的编程技能即可操作系统。这不仅提升了系统的可用性和用户体验，还使得检测过程更加直观透明，便于结果的实时观察和分析。此外，GUI还可以集成其他功能，如检测结果的保存与导出、检测参数的调整，从而为用户提供一个全面、综合的检测工作环境，促进智能检测技术的广泛应用。
软件初始界面：
运行首页
检测结果如下图：
运行结果
三、软件功能介绍
软件主要功能
-
可用于实际场景中水面目标物检测，分为14个检测类别：["bridge", "ship", "boat", "ball", "rubbish", "rock", "buoy", "platform", "habor", "mast", "tree", "animal", "grass", "person"]
-
支持图片、视频及摄像头进行检测，同时支持图片的批量检测；
-
界面可实时显示目标位置、目标总数、置信度、用时等信息;
-
支持图片或者视频的检测结果保存；
界面参数说明
置信度阈值： 也就是目标检测时的conf参数，只有检测出的目标置信度大于该值，结果才会显示；
交并比阈值： 也就是目标检测时的iou参数，只有目标检测框的交并比大于该值，结果才会显示；
线宽： 表示在绘制结果时，绘制框的线宽；
Delay： 代表运行时展示的延长时间；
检测结果说明
标签名称与置信度： 表示检测图片上标签名称与置信度；
总目标数： 表示画面中检测出的目标数目；
目标选择： 可选择单个目标进行位置信息、置信度查看。
目标位置： 表示所选择目标的检测框，左上角与右下角的坐标位置。
主要功能说明
功能视频演示见视频，以下是简要的操作描述。
-
（1） 图片检测说明
点击打开图片按钮，选择需要检测的图片，或者点击打开文件夹按钮，选择需要批量检测图片所在的文件夹，操作演示如下：点击目标下拉框后，可以选定指定目标的结果信息进行显示。点击保存按钮，会对检测结果进行保存，存储路径为：runs目录下。
-
（2）视频检测说明
点击视频按钮，打开选择需要检测的视频，就会自动显示检测结果，再次点击可以关闭视频。点击保存按钮，会对视频检测结果进行保存，存储路径为：runs目录下。
-
（3）摄像头检测说明
点击打开摄像头按钮，可以打开摄像头，可以实时进行检测，再次点击，可关闭摄像头。
-
（4）保存图片与视频检测说明
点击保存按钮后，会将当前选择的图片【含批量图片】或者视频的检测结果进行保存,方便进行查看与后续使用。检测的图片与视频结果会存储在runs目录下。
四、数据集介绍、准备
通过网络上搜集关于实际场景中水上目标物的相关图片，并使用Labelimg标注工具对每张图片进行标注，分14个检测类别，分别时"bridge", "ship", "boat", "ball", "rubbish", "rock", "buoy", "platform", "habor", "mast", "tree", "animal", "grass", "person"， 最终数据集一共包含7647张图片，其中训练集包含5973张图片，验证集包含746张图片，测试图片748张。部分图像及标注如下图所示：
数据集各类别数目分布如下：
五、YOLOv8/YOLOv9/YOLOv10简介
YOLO（You Only Look Once）是一种流行的计算机视觉算法，用于实现实时对象检测。它由Joseph Redmon等人首次在2015年提出，并随后进行了多次改进。YOLO的核心思想是将整个图像划分为一个固定数量的格子（grid cells），然后在每个格子内同时预测多个边界框（bounding boxes）和类别概率。
YOLOv8、YOLOv9、YOLOv10是YOLO系列中最前沿的3个系列版本，他们均是基于先前YOLO版本在目标检测任务上的成功，对模型结构进行不断地优化改进，从而不断提升了性能和灵活性，在精度和速度方面都具有尖端性能。
上图是前沿的SOTA目标检测模型在经典的COCO数据集上的性能表现对比。从上图可以看出，YOLOv8、YOLOv9、YOLOv10都有较好的性能表现。下面详细介绍各个版本的相关信息。
YOLOv8的简介
源码地址：https://github.com/ultralytics/ultralytics Yolov8是一个SOTA模型，它建立在Yolo系列历史版本的基础上，并引入了新的功能和改进点，以进一步提升性能和灵活性，使其成为实现目标检测、图像分割、姿态估计等任务的最佳选择。其具体创新点包括一个新的骨干网络、一个新的Ancher-Free检测头和一个新的损失函数，可在CPU到GPU的多种硬件平台上运行。
YOLOv8网络结构如下：
版本更新对比如下：
YOLOv8创新点：
Yolov8主要借鉴了Yolov5、Yolov6、YoloX等模型的设计优点，其本身创新点不多，偏重在工程实践上，具体创新如下：
-
提供了一个全新的SOTA模型（包括P5 640和P6 1280分辨率的目标检测网络和基于YOLACT的实例分割模型）。并且，基于缩放系数提供了N/S/M/L/X不同尺度的模型，以满足不同部署平台和应用场景的需求。
-
Backbone：同样借鉴了CSP模块思想，不过将Yolov5中的C3模块替换成了C2f模块，实现了进一步轻量化，同时沿用Yolov5中的SPPF模块，并对不同尺度的模型进行精心微调，不再是无脑式一套参数用于所有模型，大幅提升了模型性能。
-
Neck：继续使用PAN的思想，但是通过对比YOLOv5与YOLOv8的结构图可以看到，YOLOv8移除了1*1降采样层。
-
Head部分相比YOLOv5改动较大，Yolov8换成了目前主流的解耦头结构(Decoupled-Head)，将分类和检测头分离，同时也从Anchor-Based换成了Anchor-Free。
-
Loss计算：使用VFL Loss作为分类损失(实际训练中使用BCE Loss)；使用DFL Loss+CIOU Loss作为回归损失。
-
标签分配：Yolov8抛弃了以往的IoU分配或者单边比例的分配方式，而是采用Task-Aligned Assigner正负样本分配策略。
YOLOv8不同模型尺寸信息：YOLOv8提供了5种不同大小的模型尺寸信息，详情如下：
| Model | size(pixels) | mAPval 50-95 | params (M) | FLOPs (B) |
|---|---|---|---|---|
| YOLOv8n | 640 | 37.3 | 3.2 | 8.7 |
| YOLOv8s | 640 | 44.9 | 11.2 | 28.6 |
| YOLOv8m | 640 | 50.2 | 25.9 | 78.9 |
| YOLOv8l | 640 | 52.9 | 43.7 | 165.2 |
| YOLOv8x | 640 | 53.9 | 68.2 | 257.8 |
一般来说,选择模型大小的原则如下: 数据集小(几百张图片):使用yolov8n或yolov8s。过大模型会过拟合。数据集中等(几千张图片):yolov8s或yolov8m。能获得较高精度,不易过拟合。数据集大(几万张图片):yolov8l或yolov8x。模型容量大,充分拟合大数据量,能发挥模型效果。超大数据集(几十万张以上):首选yolov8x。超大模型才能处理海量数据并取得最优效果.
YOLOv9简介
论文地址：https://arxiv.org/abs/2402.13616
源码地址：https://github.com/WongKinYiu/yolov9
YOLOv9在YOLOv8的网络基础上做了进一步创新，其主要专注于解决深度神经网络中信息丢失带来的挑战。信息瓶颈原理和可逆函数的创新使用是其设计的核心，确保了 YOLOv9 保持高效率和准确性。YOLOv9引入了两项关键创新概念：可编程梯度信息 （PGI）：PGI 是 YOLOv9 中引入的一个新概念，用于解决信息瓶颈问题，确保跨深度网络层保存基本数据。这样可以生成可靠的梯度，促进准确的模型更新并提高整体检测性能。
广义高效层聚合网络（GELAN）： GELAN代表了架构的战略进步，使YOLOv9能够实现卓越的参数利用率和计算效率。它的设计允许灵活集成各种计算块，使 YOLOv9 在不牺牲速度或准确性的情况下适应广泛的应用。
YOLOv9架构创新点更强大的骨干网络： YOLOv9采用了一种新的骨干网络设计，该设计在保持计算效率的同时，增强了特征的提取能力。通过引入更深的网络层次和更复杂的连接方式，YOLOv9能够更有效地捕捉图像中的上下文信息，从而提高了对目标的识别和定位精度。
改进的检测头设计： 在检测头方面，YOLOv9进行了精心的设计和优化。它采用了多尺度特征融合的策略，使得模型能够同时关注不同大小的目标。此外，YOLOv9还引入了一种新的损失函数，以更好地平衡正负样本之间的权重，从而提高了模型的训练稳定性和检测性能。
可编程梯度信息利用： YOLOv9的一个显著创新点是它对梯度信息的利用方式。通过引入可编程的梯度信息学习策略，YOLOv9能够更有效地进行模型参数的更新和优化。这种方法不仅加速了模型的收敛速度，还有助于提高模型对复杂场景和多样化任务的适应性。
YOLOv9不同模型尺寸信息： YOLOv9同样提供了5种不同大小的模型尺寸信息，详情如下：
| Model | size (pixels) | mAPval 50-95 | mAPval 50 | params (M) | FLOPs (B) |
|---|---|---|---|---|---|
| YOLOv9t | 640 | 38.3 | 53.1 | 2.0 | 7.7 |
| YOLOv9s | 640 | 46.8 | 63.4 | 7.2 | 26.7 |
| YOLOv9m | 640 | 51.4 | 68.1 | 20.1 | 76.8 |
| YOLOv9c | 640 | 53.0 | 70.2 | 25.5 | 102.8 |
| YOLOv9e | 640 | 55.6 | 72.8 | 58.1 | 192.5 |
YOLOv10介绍
论文地址：https://arxiv.org/abs/2405.14458
源码地址：https://github.com/THU-MIG/yolov10
YOLOv10 的架构建立在以前 YOLO 模型的优势之上，通过消除非最大抑制 （NMS） 和优化各种模型组件， 实现了最先进的性能，并显著降低了计算开销。
模型网络结构由以下组件组成： 主干网：YOLOv10 中的主干网负责特征提取，使用增强版的 CSPNet（Cross Stage Partial Network）来改善梯度流并减少计算冗余。颈部：颈部被设计成聚合来自不同尺度的特征，并将它们传递到头部。它包括 PAN（路径聚合网络）层，用于有效的多尺度特征融合。一对多头：在训练过程中为每个对象生成多个预测，以提供丰富的监督信号，提高学习准确性。一对一头：在推理过程中为每个对象生成一个最佳预测，消除对 NMS 的需求，从而减少延迟并提高效率。YOLOv10创新点如下无 NMS 训练： 利用一致的双重分配来消除对 NMS 的需求，从而减少推理延迟。整体模型设计： 从效率和精度两个角度对各种组件进行全面优化，包括轻量级分类头、空间通道解耦下采样和秩引导块设计。增强的模型功能： 整合大核卷积和部分自注意力模块，可在不增加大量计算成本的情况下提高性能。
YOLOv10不同模型尺寸信息：
YOLOv10 提供6种不同的型号规模模型，以满足不同的应用需求：
| Model | Input Size | APval | params (M) | FLOPs (G) |
|---|---|---|---|---|
| YOLOv10-N | 640 | 38.5 | 2.7 | 6.7 |
| YOLOv10-S | 640 | 46.3 | 7.2 | 21.6 |
| YOLOv10-M | 640 | 51.1 | 15.4 | 59.1 |
| YOLOv10-B | 640 | 52.5 | 19.1 | 92.0 |
| YOLOv10-L | 640 | 53.2 | 24.4 | 120.3 |
| YOLOv10-X | 640 | 54.4 | 29.5 | 160.4 |
YOLOv10-N：Nano 版本，适用于资源极度受限的环境。YOLOv10-S：平衡速度和精度的小型版本。YOLOv10-M：通用的中型版本。YOLOv10-B：平衡版本，宽度增加，精度更高。YOLOv10-L：大版本，以增加计算资源为代价，实现更高的精度。YOLOv10-X：超大版本，可实现最大的精度和性能。
六、模型训练、评估与推理
本文主要基于YOLOv8n、YOLOv9t、YOLOv10n这3种模型进行模型的训练，训练完成后对3种模型在验证集上的表现进行全面的性能评估及对比分析。模型训练和评估流程基本一致，包括：数据集准备、模型训练、模型评估。下面主要以YOLOv8为例进行训练过程的详细讲解，YOLOv9与YOLOv10的训练过程类似。
1. 模型训练
准备好数据集后，将图片数据以如下格式放置在项目目录中。在项目目录中新建datasets目录
同时我们需要新建一个data.yaml文件，用于存储训练数据的路径及模型需要进行检测的类别。YOLOv8在进行模型训练时，会读取该文件的信息，用于进行模型的训练与验证。data.yaml的具体内容如下：
# Ultralytics YOLO 🚀, AGPL-3.0 license
# Package-seg dataset by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/segment/package-seg/
# Example usage: yolo train data=package.yaml
path: /home/test/projects/Projectpractice/yolov10-main/ultralytics/datasets/水面目标物数据集
train: images/train
val: images/valid
test: images/test
nc: 14
names:
0: bridge
1: ship
2: boat
3: ball
4: rubbish
5: rock
6: buoy
7: platform
8: habor
9: mast
10: tree
11: animal
12: grass
13: person
注：train与val后面表示需要训练图片的路径，建议直接写自己文件的绝对路径。数据准备完成后，通过调用train.py文件进行模型训练，epochs参数用于调整训练的轮数，batch参数用于调整训练的批次大小【根据内存大小调整，最小为1】，代码如下：
模型常用训练超参数参数说明：YOLOv8 模型的训练设置包括训练过程中使用的各种超参数和配置。这些设置会影响模型的性能、速度和准确性。关键的训练设置包括批量大小、学习率、动量和权重衰减。此外，优化器、损失函数和训练数据集组成的选择也会影响训练过程。对这些设置进行仔细的调整和实验对于优化性能至关重要。以下是一些常用的模型训练参数和说明：
| Argument | Default | Description |
|---|---|---|
| model | None | 模型或者模型配置文件 |
| data | None | 数据集配置文件 |
| epochs | 100 | 训练的轮次 |
| patience | 100 | 准确率为提升停止的轮次 |
| batch | 16 | 训练的批次大小 |
| imgsz | 640 | 图像的尺寸 |
| save | True | 是否需要保存训练的结果与模型 |
| device | None | 训练设备 |
| workers | 8 | 多线程数据集加载 |
| project | None | 项目名称 |
| exist_ok | False | 是否覆盖现有试验 |
| pretrained | True | 是否使用预训练模型 |
| optimizer | 'auto' | 优化器选择 |
| verbose | False | 是否详细输出 |
| seed | 0 | 种子数 |
| deterministic | True | 是否启动确定性模式 |
| rect | False | 单类别模型训练 |
| cos_lr | False | 是否使用余弦学习率调度器 |
| resume | False | 是否从最近的训练断掉权重继续训练 |
| amp | True | 是否开启混合精度训练 |
2.训练结果评估
在深度学习中，我们通常用损失函数下降的曲线来观察模型训练的情况。YOLOv8在训练时主要包含三个方面的损失：定位损失(box_loss)、分类损失(cls_loss)和动态特征损失（dfl_loss），在训练结束后，可以在runs/目录下找到训练过程及结果文件，如下所示：
各损失函数作用说明：
定位损失box_loss： 预测框与标定框之间的误差（GIoU），越小定位得越准；
分类损失cls_loss： 计算锚框与对应的标定分类是否正确，越小分类得越准；
动态特征损失（dfl_loss）： DFLLoss是一种用于回归预测框与目标框之间距离的损失函数。在计算损失时，目标框需要缩放到特征图尺度，即除以相应的stride，并与预测的边界框计算Ciou Loss，同时与预测的anchors中心点到各边的距离计算回归DFLLoss。 主要工作内容
本文训练结果如下：
我们通常用PR曲线来体现精确率和召回率的关系，本文训练结果的PR曲线如下。mAP表示Precision和Recall作为两轴作图后围成的面积，m表示平均，@后面的数表示判定iou为正负样本的阈值。mAP@.5：表示阈值大于0.5的平均mAP，可以看到本文模型目标检测的mAP@0.5值为0.849，结果还是很不错的。
3.模型推理
模型训练完成后，我们可以得到一个最佳的训练结果模型best.pt文件，在runs/train/weights目录下。我们可以使用该文件进行后续的推理检测。图片检测代码如下：
#coding:utf-8
from ultralytics import YOLO
import cv2
# 所需加载的模型目录
path = 'models/best.pt'
# 需要检测的图片地址
img_path = "TestFiles/Burr16_png_jpg.rf.75fc417fe0f65aa2ff850971806cddd7.jpg"
# 加载预训练模型
model = YOLO(path, task='detect')
# 检测图片
results = model(img_path)
print(results)
res = results[0].plot()
# res = cv2.resize(res,dsize=None,fx=2,fy=2,interpolation=cv2.INTER_LINEAR)
cv2.imshow("YOLOv8 Detection", res)
cv2.waitKey(0)
检测结果如下：
七、YOLOv8/YOLOv9/YOLOv10性能对比分析
本文在介绍的数据集上分别训练了YOLOv8n、YOLOv9t、YOLOv10n这3种模型用于对比分析，训练轮数为150个epoch。主要分析这3种模型的训练结果在Precision（精确度）、Recall（召回率）、mAP50、mAP50-95、F1-score等性能指标上的表现，以选出更适合本数据集的最优模型。3种模型基本信息如下：
| Model | size (pixels) | mAPval 50-95 | params (M) | FLOPs (B) |
|---|---|---|---|---|
| YOLOv8n | 640 | 37.3 | 3.2 | 8.7 |
| YOLOv9t | 640 | 38.3 | 2.0 | 7.7 |
| YOLOv10n | 640 | 38.5 | 2.7 | 6.7 |
FlOPs(floating point operations)：浮点运算次数，用于衡量算法/模型的复杂度。params (M):表示模型的参数量
-
各模型性能评估
在YOLOv8、YOLOv9、YOLOv10这3种模型训练完成后，我们可以通过验证集对各个模型分别进行性能评估。YOLOv8模型在验证集上的性能评估结果如下：
Class Images Instances Box(P R mAP50 mAP50-95)
all 746 2154 0.832 0.805 0.831 0.471
bridge 746 190 0.926 0.984 0.971 0.735
ship 746 376 0.92 0.928 0.953 0.695
boat 746 754 0.895 0.859 0.908 0.549
ball 746 341 0.869 0.501 0.715 0.316
rubbish 746 65 0.779 0.708 0.759 0.415
rock 746 226 0.856 0.757 0.828 0.382
buoy 746 14 0.733 0.786 0.679 0.383
platform 746 52 0.738 0.865 0.821 0.557
mast 746 31 0.914 0.935 0.977 0.619
tree 746 17 0.737 0.882 0.785 0.425
animal 746 6 0.842 0.891 0.948 0.24
grass 746 6 0.794 0.667 0.669 0.38
person 746 76 0.816 0.701 0.789 0.431
Speed: 0.3ms preprocess, 5.6ms inference, 0.0ms loss, 4.1ms postprocess per image
Results saved to runs/detect/val3
💡 Learn more at https://docs.ultralytics.com/modes/val
Class:表示模型的检测类别名称；Images:表示验证集图片数目；Instances:表示在所有图片中目标数；P:表示精确度Precison; R:表示召回率Recall; mAP50：表示IoU（交并比）阈值为0.5时的平均精度。mAP50-95：表示从IoU为0.5到0.95的范围内【间隔0.05】，模型的平均精度。
YOLOv9模型在验证集上的性能评估结果如下：
Class Images Instances P R mAP50 mAP50-95: 100%|██████████| 24/24 00:13
all 746 2154 0.848 0.798 0.851 0.498
bridge 746 190 0.923 0.984 0.978 0.736
ship 746 376 0.933 0.934 0.959 0.705
boat 746 754 0.882 0.874 0.915 0.567
ball 746 341 0.83 0.531 0.74 0.336
rubbish 746 65 0.812 0.729 0.842 0.441
rock 746 226 0.871 0.783 0.847 0.418
buoy 746 14 0.623 0.786 0.821 0.417
platform 746 52 0.791 0.799 0.821 0.571
mast 746 31 0.936 0.936 0.989 0.669
tree 746 17 0.869 0.882 0.91 0.566
animal 746 6 0.812 0.728 0.734 0.187
grass 746 6 0.908 0.667 0.668 0.414
person 746 76 0.836 0.74 0.844 0.447
YOLOv10n模型在验证集上的性能评估结果如下：
Class Images Instances Box(P R mAP50 mAP50-95): 100%|█████████
all 746 2154 0.8 0.758 0.818 0.457
bridge 746 190 0.876 0.969 0.977 0.705
ship 746 376 0.91 0.911 0.951 0.676
boat 746 754 0.875 0.861 0.9 0.545
ball 746 341 0.807 0.514 0.7 0.299
rubbish 746 65 0.746 0.646 0.749 0.369
rock 746 226 0.877 0.758 0.829 0.378
buoy 746 14 0.655 0.68 0.731 0.288
platform 746 52 0.745 0.787 0.793 0.521
mast 746 31 0.87 0.903 0.957 0.614
tree 746 17 0.595 0.647 0.803 0.543
animal 746 6 0.829 0.812 0.77 0.23
grass 746 6 0.77 0.667 0.689 0.377
person 746 76 0.842 0.697 0.781 0.391
Speed: 1.0ms preprocess, 1.5ms inference, 0.0ms loss, 0.0ms postprocess per image
整体来看，在此数据集上YOLOv8n在各个指标方面都表现略微比YOLOv9t与YOLOv10n要好一点。因此推荐使用YOLOv8n训练的模型进行此类的检测任务。
八、可视化系统制作
基于上述训练出的目标检测模型，为了给此检测系统提供一个用户友好的操作平台，使用户能够便捷、高效地进行检测任务。博主基于Pyside6开发了一个可视化的系统界面，通过图形用户界面（GUI），用户可以轻松地在图片、视频和摄像头实时检测之间切换，无需掌握复杂的编程技能即可操作系统。
1. Pyside6简介
PySide6是一个用于创建跨平台GUI（图形用户界面）应用程序的库，它是Qt for Python的官方库。Qt本身是一个跨平台的C++应用程序框架，广泛用于开发具有图形用户界面(GUI)的应用程序。PySide6允许开发者使用Python语言访问Qt的功能，从而简化了开发过程
下面对Pyside6 的基本原理进行详细介绍：
1.基本架构
PySide6是用于Qt应用程序的Python绑定。基于Qt框架，它提供了一套强大的GUI工具集，适用于开发跨平台的桌面应用程序。一个基本的PySide6应用程序通常包含以下组件：导入QtWidgets模块，它是创建用户界面的基本模块；创建一个应用程序对象；创建窗口部件，如按钮、标签等；通过信号和槽机制连接窗口部件的事件；显示并运行应用程序。
2.事件驱动编程
Pyside6基于事件驱动编程模型，主要通过信号（signals）和槽（slots）机制实现用户与应用程序之间的交互。当用户与 GUI 进行交互（如点击按钮、调整滑块等）时，会触发信号，这些信号可以连接到槽函数或方法，以执行特定操作。
3.Qt对象模型
Pyside6的核心是 Qt 对象模型，所有的控件和窗口部件都是从 QObject 类派生而来的。它们拥有复杂的父子关系，确保父对象在销毁时自动销毁所有子对象，避免内存泄漏。
4.部件（Widgets）
Pyside6提供了丰富的内置部件，如按钮、标签、文本框、表格、树、标签页等，几乎涵盖了所有常见的 GUI 控件。这些部件可以直接使用，也可以通过继承进行自定义。
5.布局管理器
提供了强大的布局管理功能，可以通过 QLayout 和其子类（如 QHBoxLayout, QVBoxLayout, QGridLayout）来控制部件在窗口内的摆放方式。这使得界面的设计变得灵活且易于维护。
6.资源管理
Pyside6支持资源文件管理，可以将图像、图标、样式表等资源打包进应用程序中。资源文件通常以 .qrc 格式存储，并通过资源管理器集成到应用程序中。
7.信号与槽机制
Pyside6信号与槽机制是 Qt 框架的核心特性之一，它允许对象之间进行松耦合通信。通过信号可以触发槽函数来处理各种事件，使代码逻辑更加清晰和模块化。
8.跨平台性
Pyside6是跨平台的，支持 Windows、Mac 和 Linux 等操作系统，编写一次代码即可运行在多个平台上。
关于该系统涉及到的完整源码、UI界面代码、数据集、训练代码、训练好的模型、测试图片视频等相关文件，均已打包上传，感兴趣的小伙伴可以通过下载链接自行获取。
视频演示https://www.bilibili.com/video/BV1RseMetE11/?vd_source=57a174b19efc43c207675a3a9964ec71
注意：该代码基于Python3.8开发，运行界面的主程序为Main.py，其他测试脚本说明见上图。为确保程序顺利运行，请按照程序运行说明文档txt配置软件运行所需环境。
获取方式 关注下方公众号：【AI算法爱好者角落】，查询即可获取下载方式
知乎解析：AI探索先锋
扫一扫
200
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
