技術(shù)特征:1.一種用于衛(wèi)星影像船舶檢測的改進(jìn)yolov8模型,其特征在于,包括以下步驟:
2.根據(jù)權(quán)利要求1所述的改進(jìn)yolov8模型,其特征在于,所述步驟(2)的特征提取模塊具體如下:
3.根據(jù)權(quán)利要求2所述的改進(jìn)yolov8模型,其特征在于,所述步驟(2)的特征融合模塊具體如下:
4.根據(jù)權(quán)利要求3所述的改進(jìn)yolov8模型,其特征在于,所述步驟(2)的檢測輸出模塊具體如下:
5.根據(jù)權(quán)利要求1所述的改進(jìn)yolov8模型,其特征在于,所述步驟(3)的改進(jìn)邊界框損失函數(shù)具體如下:
技術(shù)總結(jié)本發(fā)明公開了一種用于衛(wèi)星影像船舶檢測的改進(jìn)YOLOv8模型,包括以下步驟:(1)數(shù)據(jù)準(zhǔn)備:輸入遙感衛(wèi)星影像,對影像進(jìn)行數(shù)據(jù)預(yù)處理,而后裁剪切片數(shù)據(jù)并劃分為訓(xùn)練集、驗證集;(2)模型改進(jìn):改進(jìn)YOLOv8模型的特征提取模塊增加輸出一個尺度更小的圖像特征;特征融合模塊使用了尺度更小的圖像特征;檢測輸出模塊可以提取尺度更小的目標(biāo);(3)模型訓(xùn)練:訓(xùn)練時對輸入的切片數(shù)據(jù)采取增強手段;改進(jìn)邊界框損失函數(shù)的增加部分對應(yīng)小目標(biāo)檢測分支的輸出;(4)模型推理:對衛(wèi)星影像裁剪得到待測切片,待測切片經(jīng)過改進(jìn)YOLOv8模型,獲得目標(biāo)檢測結(jié)果。本發(fā)明的改進(jìn)YOLOv8模型可有效檢測小目標(biāo)船舶,且檢測精度高;同時,能夠有效應(yīng)對衛(wèi)星影像特有的圖像質(zhì)量問題。
技術(shù)研發(fā)人員:吳祖勇,王應(yīng)博,王詒琬,陳木森,王清蓮,朱濟帥
受保護(hù)的技術(shù)使用者:海南長光衛(wèi)星信息技術(shù)有限公司
技術(shù)研發(fā)日:技術(shù)公布日:2024/12/19