cornernet------loss

首先我們使用網絡訓練出左上角的heatmap_tl以及右下角的hetmap_br,這個heatmap維度爲(batch,128,128,80)其中80是類別數。在製作數據集的時候,將數據也同樣製作成了(batch,128,128,80)。以左上角爲例,我們每一類都有一個(128,128)的矩陣與之對應。輸入網絡一張圖片,我們會得到左上角以及右下角的heatmap,這個heatmap裏面又包括了每一
本站公眾號
   歡迎關注本站公眾號,獲取更多信息