船舶检测模型训练方法、装置、电子设备及存储介质

    公开(公告)号:CN119169402A

    公开(公告)日:2024-12-20

    申请号:CN202311515863.8

    申请日:2023-11-13

    Abstract: 本申请提供一种船舶检测模型训练方法、装置、电子设备及存储介质,涉及计算机处理技术领域。方法包括:获取用于模型训练的数据集;将数据集中的样本图像输入待训练的初始检测模型中的骨干网络进行多尺度特征提取,得到每个样本图像对应的不同尺度的多个尺度特征图;将多个尺度特征图输入初始检测模型中的特征提取网络,得到第一特征图和第二特征图;将各个第一特征图和第二特征图输入检测头,得到样本图像对应的检测结果,最后根据检测结果对初始检测模型进行参数调整和迭代训练,直至初始检测模型满足预设模型收敛条件。如此,改善传统船舶检测方式存在小目标检测结果不准确、易漏检的问题。

Patent Agency Ranking