如何使用SSD Model 來pretrain 其他data
2020/01/22 下午 02:59
電腦視覺深度學習討論版
cleo
觀看數：17
回答數：2
收藏數：0
cvdl-1-d29
如果我想使用這個SSD Model
來 pretrain 我額外製作的data 而且label 不屬於coco之中的80類 要怎麼做呢
目前我有labelimg 後的XML檔 與 原圖 但不知道如何使用
回答列表
-
2020/01/22 下午 04:52Jeffrey贊同數：0不贊同數：0留言數：1
-
2020/01/23 下午 06:59楊哲寧贊同數：0不贊同數：0留言數：1
您好，具體步驟如下：
1.調整輸出層類別預測feature Map深度，變成類別+背景(ex. Coco 是 80+1, Voc是20+1)
2.準備dataLoader，要確保輸入形式符合原本格式，因為您的資料與COCO存放的格式不相同，需要不一樣的預處理取出資訊，舉個例子，輸入影像要變為(batch_size,channel,height,width)，並且height,width==300(可以調整，但會比較複雜，default box 也要改)，並且經過Normalization，label也要轉換成程式導讀中的範例格式，dataLoader可參考這篇：https://blog.csdn.net/guyuealian/article/details/88343924
3. Pretrained weight 要使用的話，最後幾層的weight不能載入，因為大小不符合。