Keras vgg轉移學習保存/加載自定義完全連接圖層

Question

在轉移學習中，只能在訓練期間保存完全連接的圖層，並在以後加載以便繼續訓練？如果可能的話，該怎麼做？

Answer 1

這可能不是最好的解決方案，但我認爲這可以解決您的問題。

import pickle 

#set a name for all fully connected layers. 
model.add(Dense(...,name='fc1')) 
model.add(Dense(...,name='fc2')) 
model.add(Dense(...,name='fc3')) 


layers_to_save = ['fc1','fc2','fc3'] # add here any layer you want to save 

# Save weights to a dictionary 
weights_dict = dict([(layer.name, layer.get_weights()) for layer in model.layers if layer.name in layers_to_save]) 

with open('filename.pickle', 'wb') as handle: 
    pickle.dump(weights_dict, handle, protocol=pickle.HIGHEST_PROTOCOL) 




# Load weights 
with open('filename.pickle', 'rb') as handle: 
    weights_dict = pickle.load(handle) 

for name in layers_to_save: 
    model.get_layer(name).set_weights(weights_dict[name]) 

您還可以檢查此keras blog post下半年的另一種方法。