SSD目標檢測訓練自己的數據集:計算數據集當中每一個類別的框的數量

我在看有個同行的博客,發現他把SSD的訓練總結的非常好,我用他的訓練方法,使用VOC2007數據集進行訓練跑通了代碼。環境是WINDOWS10+tensorflow-gpu1.9版本。但是這個大兄弟的最終的類別框的計算代碼有問題。這裏把我寫的代碼與諸位分享一下。針對的是VOC2007的20分類數據集,最初發現我的輸出結果與那個仁兄的訓練數據對不上,我仔細檢查了areoplane類別的計算結果,發現
相關文章
相關標籤/搜索