-
公开(公告)号:CN119205810A
公开(公告)日:2024-12-27
申请号:CN202411261011.5
申请日:2024-09-10
Applicant: 湖南大学
IPC: G06T7/11 , G06T7/00 , G06N3/0464 , G06N3/0455 , G06N3/0985
Abstract: 本申请涉及一种基于神经网络架构搜索的工业缺陷分割方法,包括:通过跨尺寸权重共享策略使权重初始化的卷积神经网络、Transformer、多层感知机在不同尺寸下互相共享权重;构建搜索空间,将权重共享处理后的卷积神经网络、Transformer、多层感知机放入搜索空间;搜索空间用于同时搜索三种网络;基于搜索空间中三种网络的所有组合构建超网络,并训练超网络;对训练好的超网络进行进化搜索,得到最优子网络;基于预处理的工业产品表面缺陷数据集训练最优子网络,基于训练好的最优子网络构建缺陷分割模型;将待检测图像输入至缺陷分割模型输出像素级分割结果。该方法解决了传统检测方法在精度、效率和适应性方面的不足。