一种水雾船舶检测方法、系统及计算机存储介质

专利主权项内容

1.一种水雾船舶检测方法，其特征在于，包括：S1、获取训练集，所述训练集分为多组，每组中包括一张清晰船舶图片和对应的一张水雾船舶图片；将所述训练集中全部图片的船舶类别和坐标进行人工标定，得到人工标定结果；S2、将训练集中当前组中当前张清晰船舶图片输入到清晰船舶检测模型的主干网络中，得到当前每个阶段清晰特征图；将所述当前每个阶段清晰特征图输入到所述清晰船舶检测模型的检测头中，得到当前清晰预测结果；S3、将训练集中当前组中当前张水雾船舶图片依次输入到水雾船舶检测模型的主干网络、去雾网络中，得到当前张伪清晰船舶图片；将当前张伪清晰船舶图片输入到所述水雾船舶检测模型的主干网络中，得到当前每个阶段伪清晰特征图；将所述当前每个阶段伪清晰特征图输入到所述水雾船舶检测模型的检测头中，得到当前伪清晰预测结果；S4、根据所述当前清晰预测结果和当前张清晰船舶图片的人工标定结果计算得到当前标定清晰损失值，根据所述当前标定清晰损失值反向更新所述清晰船舶检测模型；根据所述当前张清晰船舶图片、所述当前张伪清晰船舶图片计算得到当前像素损失值；根据所述当前每个阶段清晰特征图、所述当前每个阶段伪清晰特征图计算得到当前所有阶段特征损失值；根据所述当前伪清晰预测结果和当前张水雾船舶图片的人工标定结果计算得到当前标定伪清晰损失值；根据所述当前伪清晰预测结果和当前清晰预测结果计算得到当前指导伪清晰损失值；将所述当前像素损失值、所述当前所有阶段特征损失值、所述当前标定伪清晰损失值、所述当前指导伪清晰损失值求和得到当前水雾损失值，根据所述当前水雾损失值反向更新所述水雾船舶检测模型；S5、重复S2~S4，直至将训练集中所有组的清晰船舶图片和对应的水雾船舶图片训练完并进行多轮训练后，得到目标水雾船舶检测模型；S6、将待检测的水雾船舶图片输入到所述目标水雾船舶检测模型中进行检测，得到船舶位置和类别。