一种基于YOLOv5改进的口罩佩戴检测方法

    公开(公告)号:CN116453186A

    公开(公告)日:2023-07-18

    申请号:CN202310400564.3

    申请日:2023-04-14

    申请人: 淮阴工学院

    摘要: 本发明涉及一种基于YOLOv5改进的口罩佩戴检测方法,属于目标检测领域。该方法包含以下步骤:挖掘目标图像样本数据,构建用于口罩佩戴检测的样本数据集;分析口罩佩戴检测存在的问题,针对问题对目标检测模型YOLOv5进行改进。采用K‑means++算法代替K‑means算法来获取anchor参数,优化目标锚框;添加注意力模块ACmix以及将原始YOLOv5网络颈部结构替换为加权双向特征金字塔网络(BiFPN),优化对小目标的检测效果,提升目标检测精确度;使用结合标准卷积和深度可分离卷积并进行通道清洗的鬼影混洗卷积(GSConv)模块替换传统卷积模块,提升网络速度;引入SIoU_Loss作为基于YOLOv5改进算法损失函数的Bounding Box Regression Loss,使得损失函数平稳收敛,降低预测误差,提升回归精度。本发明算法采用改进YOLOv5检测算法不仅提升了口罩佩戴检测精确度,还改善对远距离以及密集人群处佩戴口罩小目标检测效果,降低了误检漏检情况。

    一种基于EIEM边缘特征提取的钢铁缺陷检测方法

    公开(公告)号:CN118823473A

    公开(公告)日:2024-10-22

    申请号:CN202410959754.3

    申请日:2024-07-17

    申请人: 淮阴工学院

    摘要: 本发明公开一种基于EIEM边缘特征提取的钢铁缺陷检测方法,首先设计一个边缘信息提取模块EIEM替换C2f中的BottleNeck,该模块采用sobel算子来提取特征的边缘信息并与整体空间信息进行特征融合,获得更加丰富的语义信息;其次,替换上采样方法为CARAFE,该方法通过编码器能够学习到适合自己特征信息的卷积核,以指导更加精确的特征重建;最后,将原来的检测头换成了动态统一的检测头(DyHead),以应对不同缺陷的尺寸和变换位置,而且正因为这个检测头,在不同的尺寸、位置、任务中,都能动态的适应,不需要改变检测头。与现有技术相比,本发明克服了现有技术中鲁棒性差、引入噪声导致缺陷检测准确率低的问题,利用改进YOLOv8s模型架构,具有准确性高,适用广度大的优点。

    一种基于GS_YOLOv7的火灾检测方法
    4.
    发明公开

    公开(公告)号:CN117292216A

    公开(公告)日:2023-12-26

    申请号:CN202310400563.9

    申请日:2023-04-14

    申请人: 淮阴工学院

    摘要: 本发明公开了一种基于GS_YOLOv7的火灾检测方法,包括改进了模型Backbone与Head部分,引入了Shuffle Attention Module;引入了GSConv模块;在Head层中引入VoVGSCSP module;在训练策略方面,将YOLOv7网络模型CBS模块改进为效率更高的CBF模块;将模型损失函数改为EIoU loss,一张640*640的特征图输入到网络的Backbone层,经过CBF,E‑ELAN以及MP模块后输出三张不同的特征图,将20*20*1024大小的特征图输入到Head层经过VoVGSCSP module中可提高网络的推理速度,特征图经过GSConv模块,模型可以提取更加丰富的语义信息,在特征提取后的非线性激活阶段,Funnel激活函数增强了模型的空间感知能力,引入了EIoU loss使得模型的收敛速度更快。本发明通过优化模型结构,引入新的激活函数和损失函数,提高了模型的效率和泛用性。

    一种基于迁移学习和改进YOLOv5电力工人安全帽佩戴检测方法

    公开(公告)号:CN117275041A

    公开(公告)日:2023-12-22

    申请号:CN202311286008.4

    申请日:2023-09-28

    申请人: 淮阴工学院

    摘要: 本发明公开了一种基于迁移学习和改进YOLOv5电力工人安全帽佩戴检测方法,具体包含以下步骤:构建施工场景下的安全帽佩戴检测数据集A,构建电力作业场景下的安全帽佩戴检测数据集B;改进电力作业场景下安全帽佩戴检测模型YOLOv5,将YOLOv5主干网络替换为基于动态卷积ODConv的ODConv_MobileNetv2主干网络,提取小目标安全帽特征信息效率大幅上升并减少额外的计算量参数;在颈部网络结构中,添加基于跨空间学习的高效多尺度注意力EMA模块,减少参数量的同时提升小目标安全帽佩戴检测的精度;采用一种基于最小点距离的IoU损失MPDIoU作为改进YOLOv5的损失函数,本发明可以提升小样本电力作业场景数据集下的小目标安全帽佩戴检测精度,降低电力作业场景下的安全帽佩戴检测漏检和误检问题。

    基于改进YOLOv5s网络的热轧钢带表面缺陷检测方法

    公开(公告)号:CN117132827A

    公开(公告)日:2023-11-28

    申请号:CN202311105750.0

    申请日:2023-08-30

    申请人: 淮阴工学院

    摘要: 本发明公开了一种基于改进YOLOv5s网络的热轧钢带表面缺陷检测方法,该方法使用YOLOv5s目标检测模型作为基础,在此基础上提出一种改进的YOLOv5s网络模型。首先,将neck部分的结构改换为AFPN结构,实验结果表明该结构与其他最先进的特征金字塔网络相比较更具竞争力,而且每秒浮点运算(FLOPs)也最低,并在较低参数下达到了比Baseline更好的性能。其次,使用GIoU损失函数,其不仅关注重合区域,还关注其他非重合区域,更好的反映了两者的重合程度,并且当IoU=0时,两个框之间的距离仍然很好的表示出来。最后,添加注意力机制模块,选用EffectiveSE卷积神经网络结构加入YOLOv5s网络中,实验表明它与SENet等经典网络结构相比分类精度更高、收敛速度更快。本发明所提出的方法在原始网络基础上有效提升了模型对热轧钢带表面缺陷检测的性能,降低了模型的参数量和计算量,缩短了模型的推理时间。