ssd.pytorch源碼分析(五)—損失函數及Hard negative mining

MultiBoxLoss源碼 SSD論文鏈接 本文代碼涉及很多複雜矩陣索引操作,推薦閱讀。 損失函數總覽 在SSD中,默認框default boxes和真實目標ground truth先進行匹配。 匹配策略細節 然後根據匹配到的一對boxes分別計算分類損失和定位損失。 從上面的描述可以看出,可能有多個default boxes匹配到一個ground truth的情況。其中α爲權重係數,論文和代碼
相關文章
相關標籤/搜索