YOLOv4 的 Keras 實作
在 YOLOv4 的這個 Keras 實作中,是否有可能以某種方式從最后保存的最佳權重繼續訓練?類似于以下內容:
model_checkpoint_callback = tf.keras.callbacks.ModelCheckpoint(
filepath=checkpoint_filepath,
monitor='val_binary_accuracy',
mode='max',
save_best_only=True)
model.load_weights(checkpoint_filepath)
uj5u.com熱心網友回復:
根據這些行,存盤庫會自動處理路徑上的權重;因此,要加載一個預訓練的權重(.h5檢查點或.weights進行遷移學習,其余部分遵循訓練筆記本;
model = Yolov4(weight_path='mytraining.weights',
class_name_path=class_name_path)
轉載請註明出處,本文鏈接:https://www.uj5u.com/caozuo/455309.html
