logo
Loading...

yolo權重檔案轉換問題 - Cupoy

專家你們好!我是使用這位大神的keras版YOLO,https://github.com/qqwwe...

yolo

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