logo
Loading...

想請問 yolo v3 的訓練代碼 相關問題 - Cupoy

您好:想問幾個問題1. 再訓練「Train with frozen layers first, to...

yolo

想請問 yolo v3 的訓練代碼 相關問題

2020/05/16 09:12 上午
電腦視覺深度學習討論版
蕭聖哲
觀看數:8
回答數:1
收藏數:0
yolo

您好:

想問幾個問題


1. 

再訓練「Train with frozen layers first, to get a stable loss」這個步驟的時候,如果電腦突然斷掉了,假設我訓練到「Epoch 40」,我再訓練過程中也得到了Epoch 40」的h5檔案,我是否可以把這個訓練到一半的h5檔案當作初始權重,於weights_path」放入其路徑,重新訓練呢

create_model(input_shape, anchors, num_classes,
            freeze_body
=2, weights_path='XXXX.h5')


2. 

如果我把「Epoch 40」的h5檔案,於weights_path」放入其路徑重新訓練,我在「initial_epoch」依然要維持0嗎,還是從40開始就可以了

model.fit_generator(data_generator_wrapper(lines[:num_train], batch_size, input_shape, anchors, num_classes),
           steps_per_epoch
=max(1, num_train//batch_size),
           validation_data
=data_generator_wrapper(lines[num_train:], batch_size, input_shape, anchors, num_classes),
           validation_steps
=max(1, num_val//batch_size),
           epochs
=50,
           
initial_epoch=0,
           callbacks
=[logging, checkpoint])


3. 

假設我訓練到「Epoch 30」驗證集的loss開始不降反而微微升高,理論上第一階段的訓練(Train with frozen layers first, to get a stable loss」)是否是直接拿Epoch 30」的h5檔案進入第二階段的訓練(「Unfreeze and continue training, to fine-tune)

然後也想問,第一階段與第二階段的訓練差異在哪呢,為何要把所有的把所有 layer 都改為 trainable


4.

另外在資料格式的部分看有兩種情況

A. 有人是拿物件中心點的位置跟相對寬度去計算(https://chtseng.wordpress.com/2018/09/01/%e5%bb%ba%e7%ab%8b%e8%87%aa%e5%b7%b1%e7%9a%84yolo%e8%be%a8%e8%ad%98%e6%a8%a1%e5%9e%8b-%e4%bb%a5%e6%9f%91%e6%a9%98%e8%be%a8%e8%ad%98%e7%82%ba%e4%be%8b/)

B. 有人是直接拿物件左上跟右下的座標點去計算

(http://omnixri.blogspot.com/2019/11/aicolumnyolov3.html)

為何會有兩種不同差異呢


以上問題 謝謝老師的回覆!!