yolo權重檔案轉換問題 - Cupoy

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不相同。