yolo權重檔案轉換問題
2020/05/12 上午 09:33
電腦視覺深度學習討論版
陳政陽
觀看數:86
回答數:1
收藏數:0
yolo
專家你們好!
我是使用這位大神的keras版YOLO,https://github.com/qqwweee/keras-yolo3
在訓練YOLO時有遇到兩個問題想要請教:
1. 訓練YOLO前必須先把權重轉換為.h5檔案,在這裡好像有使用到yolov3.cfg這個檔案。
然後要訓練自己資料集的話需要去修改yolov3.cfg裡面的內容,所以我應該是要將yolov3.cfg檔案修改完之後再進行轉換嗎??
2. 作者文中有提到,若要使用原始的預訓練權重可以另外下載darknet53_weights和使用darknet53.cfg來轉換成.h5檔,那麼這邊提到的原始預訓練權重跟課程中使用的yolov3_weights有什麼不一樣呢???
謝謝~
回答列表
-
2020/05/16 上午 11:34楊哲寧贊同數:0不贊同數:0留言數:2
針對第一題:這邊可以直接轉沒問題,不用先去修改,除非你會更改模型架構(一般只會更改預測類別與數量),這樣轉過來的weight會無法使用。
針對第二題:兩個都是YoloV3 pretrained weight, 只是model backbone不相同。