(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210981372.1 (22)申请日 2022.08.16 (71)申请人 湖南大学 地址 410082 湖南省长 沙市岳麓区麓山 南 路麓山门 申请人 无锡市锡山区半导体先进制造创新 中心 (72)发明人 刘坚　杨德志　陈宁　 (74)专利代理 机构 北京思睿峰知识产权代理有 限公司 1 1396 专利代理师 谢建云　赵爱军 (51)Int.Cl. G06T 7/00(2017.01) G06T 7/136(2017.01) G06T 5/30(2006.01)G06T 5/00(2006.01) G06T 7/13(2017.01) G06V 10/762(2022.01) G06V 10/80(2022.01) G06V 10/82(2022.01) G06N 3/04(2006.01) G06N 3/08(2006.01) (54)发明名称 小型可插拔收发光模块底座缺陷的检测方 法 (57)摘要 本发明公开了小型可插拔收发光模块底座 缺陷的检测方法， 包括： 获取光模块底座的图像； 将光模块底 座的图像输入到目标检测模型中， 得 到光模块底座的缺陷检测结果； 其中， 目标检测 模型通过下述方式生成： 构建样本数据集， 并将 样本数据集输入预训练模型； 通过K ‑means++算 法， 生成样本数据集中每个样 本数据的多个先验 框； 将多个先验框中的每个先验框与该样本数据 所标注的真实框进行交并比计算， 将交并比的值 最大时的先验框作为预测结果； 计算预测结果和 该真实框的损失函数； 基于损失函数， 对预训练 模型进行优化， 得到目标检测模型。 本发明的技 术方案提升 了光模块底座 缺陷检测的精度。 权利要求书2页 说明书17页 附图8页 CN 115409785 A 2022.11.29 CN 115409785 A 1.一种小型 可插拔收发光模块底座 缺陷的检测方法， 所述方法包括： 获取所述光模块底座的图像； 将所述光模块底座的图像输入到目标检测模型中， 得到所述光模块底座的缺陷检测结 果； 其中， 所述目标检测模型通过 下述方式生成： 构建样本数据集， 并将所述样本数据集输入预训练模型； 通过K‑means++算法， 生成所述样本数据集中每 个样本数据的多个先验框； 将多个先验框 中的每个先验框与 该样本数据 所标注的真实框进行交并比计算， 将交并 比的值最大时的先验框作为预测结果； 计算所述预测结果和该真实框的损失函数； 基于所述损失函数， 对所述预训练模型进行优化， 得到所述目标检测模型。 2.根据权利要求1所述的方法， 其中， 所述目标检测模型所采用的特征提取网络为 GhostNet， 所述目标检测模型的特征融合网络中的有效特征层的输出后和上采样层的输出 后包括注意力机制模块。 3.根据权利要求1或2所述的方法， 在所述将所述光模块底座的图像输入到目标检测模 型中之前， 还 包括： 对所述光模块底座的图像进行 预处理； 其中， 将所述 光模块底座的图像输入到目标检测模型中， 包括： 将预处理后的光模块底座的图像输入到目标检测模型中。 4.根据权利要求3所述的方法， 其中， 所述对所述 光模块底座的图像进行 预处理， 包括： 将包含所述光模块底座的部分作为底座图像从所述 光模块底座的图像中分割出来； 对所述底座图像进行倾 斜矫正。 5.根据权利要求4所述的方法， 其中， 所述将包含所述光模块底座的部分作为底座图像 从所述光模块底座的图像中分割出来， 包括： 设置灰度分割阈值， 将所述光模块底座的图像划分为灰度值大于所述灰度分割阈值和 灰度值小于或等于所述灰度分割阈值的两 部分； 确定灰度值大于所述灰度分割阈值的部分与灰度值小于或等于所述灰度分割阈值的 部分之间的方差； 在不同的取值下遍历所述灰度分割阈值， 确定所述方差最大时灰度分割阈值的数值； 采用所述方差最大时灰度分割阈值的数值， 对所述光模块底座的图像进行二值化处 理； 根据二值化处理后的图像， 从所述 光模块底座的图像中分割出底座图像。 6.根据权利要求5所述的方法， 其中， 根据二值化处理后的图像， 从所述光模块底座的 图像中分割出底座图像， 包括： 对二值化处理后的图像进行去除噪声的处 理； 在去除噪声处 理后的图像中， 计算所述底座图像的最大外 接矩阵的坐标信息； 根据所述 坐标信息， 从所述 光模块底座的图像中分割出底座图像。 7.根据权利要求5所述的方法， 其中， 根据二值化处理后的图像， 从所述光模块底座的 图像中分割出底座图像， 包括：权　利　要　求　书 1/2 页 2 CN 115409785 A 2将二值化处理后的图像中的背景去除， 得到底座图像。 8.根据权利要求4至7中任一项所述的方法， 其中， 所述对所述底座图像进行倾斜矫正， 包括： 将所述底座图像进行二 值化处理， 得到底座的二 值图像； 将所述底座的二 值图像在多个角度上旋转变换； 在每个角度上， 确定所述底座的二 值图像在垂直方向上的投影长度； 将投影长度最小时对应的角度作为底座图像的矫 正角度； 采用所述 矫正角度对所述底座图像进行矫 正。 9.一种计算设备， 包括： 至少一个处 理器； 以及 存储器， 存储有程序指令， 其中， 所述程序指令被配置为适于由所述至少一个处理器执 行， 所述程序指令包括用于执 行如权利要求1至8中任一项所述方法的指令 。 10.一种存储有程序指令的可读存储介质， 当所述程序指令被计算设备读取并执行时， 使得所述计算设备 执行如权利要求1至8中任一项所述方法。权　利　要　求　书 2/2 页 3 CN 115409785 A 3