應用二次IoU損失函數的無錨框目標檢測方法
本發明提供一種應用二次IoU損失函數的無錨框目標檢測方法,其特征在于,包括:將待檢測圖像進行預處理再輸入至預先訓練好的目標檢測模型進行推理得到對應的圖像檢測結果,其中,目標檢測模型預先通過如下訓練得到:獲取訓練用圖像數據集以及監督信息;構建初始目標檢測模型,并將訓練用圖像數據集以及監督信息輸入至該模型;使用二次IoU損失函數求定位損失,并使用Focal Loss損失函數求分類損失;分別用定位損失以及分類損失對初始目標檢測模型的模型參數進行求導,再使用反向傳播對模型參數進行更新;判斷更新后的模型參數是否達到終止條件,當判斷為是則進入下一步,否則進入訓練過程第二步;保存更新后的模型參數并進行加載得到目標檢測模型。
復旦大學
2021-01-12