一种基于改进型SSD算法的钢缆表面缺陷检测方法

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种基于改进型SSD算法的钢缆表面缺陷检测方法
申请号:CN202411509073
申请日期:2024-10-28
公开号:CN119515789A
公开日期:2025-02-25
类型:发明专利
摘要
本发明公开了一种基于改进型SSD算法的钢缆表面缺陷检测方法,包括步骤:S1、构建钢缆表面缺陷数据集,包括若干钢缆图像、以及钢缆图像对应的缺陷类别。S2、构建钢缆表面缺陷检测神经网络,利用数据集对钢缆表面缺陷检测神经网络训练,得到钢缆表面缺陷检测模型;钢缆表面缺陷检测神经网络基于SSD网络构建,引入RestNet18作为SSD网络的骨干、单向特征融合块和改进的CBAM注意力模块、同时将Focalloss作为损失函数。S3、实时采集钢缆图像,输入至钢缆表面缺陷检测模型,获得对应的缺陷类别。本发明的一种基于改进型SSD算法的钢缆表面缺陷检测方法能够有效提升钢缆生产效率,降低人工成本。
技术关键词
SSD算法 表面缺陷检测方法 sigmoid函数 钢缆 萤火虫优化算法 输出特征 缺陷类别 全局平均池化 通道注意力机制 神经网络训练 双线性插值 模块 空洞 初始聚类中心 图像 样本