关于目标检测:无人机巡检场景小目标检测与量化加速部署方案详解
明天为大家带来的剖析是有对于如何利用基于 PP-YOLOE-SOD 的小指标检测计划,实现无人机巡检疾速精确地检测出小指标的利用案例~ 在社会活动和社会生产中,巡检是一个必不可少的环节。然而,传统的人工巡检形式存在效率低下、老本昂扬、平安危险大等问题,限度了巡检的成果和范畴。无人机巡检因其高效、灵便、平安的特点被越来越多的企业采纳。然而,如何在无人机巡检中疾速精确地检测出小指标,成为了该技术的重要难点之一。本文将介绍一种基于 PP-YOLOE-SOD 的小指标检测计划, 并分享应用 PaddleSlim 模型压缩工具和 OpenVINO 在 Intel x86平台上部署减速 PP-YOLOE+ 系列模型。 我的项目链接https://aistudio.baidu.com/aistudio/projectdetail/5036782 无人机视角指标物示意 场景难点数据量和品质有余在无人机巡检场景下,要获取大量的高质量数据是十分艰难的。此外,场景的复杂性和多变性也给数据采集和标注带来了很大的挑战。 小指标检测难度大无人机巡检场景中的小指标往往尺寸较小、形态简单,容易被周围环境烦扰而导致检测误差。 模型精度受量化误差的影响大指标检测交融了指标定位和指标分类两种工作,工作复杂度高,精度受量化的影响更大。 部署效率较低在理论利用中,模型的部署效率往往是一个重要的思考因素,特地是在无人机巡检等场景中,须要实时检测并及时反馈后果,而浮点模型因为计算量较大,推理性能不佳。 方案设计如上剖析,无人机巡检场景下因为其视角较高、画面较大、指标物较小,故比拟适宜小指标检测算法。针对小指标检测难点,百度飞桨团队基于 PP-YOLOE+ 通用检测模型,从流程和算法上进行了改良,提出了一套小指标专属检测器 PP-YOLOE-SOD (Small Object Detection)。 点击文末浏览原文疾速体验https://github.com/PaddlePaddle/PaddleDetection 对于 PaddleDetection 的技术问题欢送大家在 Github 提 issue,也欢送大家点 Star 反对咱们的工作! PP-YOLOE-SOD 针对小指标有两种检测计划,别离是基于切图拼图流程优化的小指标检测计划以及基于原图模型算法优化的小指标检测计划。同时提供了数据集主动剖析脚本,只需输出数据集标注文件,便可失去数据集统计后果,辅助判断数据集是否是小指标数据集以及是否须要采纳切图策略,同时给出网络超参数参考值。综合思考精确和时延,最终抉择速度和精度较为均衡的基于原图的小指标检测计划。 如下图标红门路,整体算法时延以及算力要求更低。 小指标检测计划 性能调优模型调优:PP-YOLOE-SOD 基于原图的小指标检测计划相比 PP-YOLOE 模型,PP-YOLOE-SOD 改良点次要包含在neck中引入Transformer全局注意力机制以及在回归分支中应用基于向量的 DFL。 引入 Transformer 全局注意力机制 Transformer 在 CV 中的利用是目前钻研较为炽热的一个方向。最早的ViT间接将图像分为多个 Patch 并退出地位 Embedding 送入 Transformer Encoder 中,加上相应的分类或者检测头即可实现较好的成果。 ViT结构图 咱们这里相似,次要退出了 Position Embedding 和 Encoder 两个模块,不同的是输出是最初一层特色图。 ...