簡體   English   中英

使用來自 Keras 應用程序的模型,無需預訓練權重

[英]Use models from Keras Applications without pretrained weights

Keras 應用程序提供了一些最流行的模型架構的實現,權重在一些最流行的數據集上進行了預訓練。 這些預定義的模型對於遷移學習問題非常方便,這些問題與模型訓練的數據集相似。

但是如果我有一個非常不同的問題並且想要在新數據集上完全訓練模型怎么辦? 如果我沒有預訓練權重,如何使用應用程序中的模型根據我自己的數據集從頭開始訓練?

您可以為權重變量分配 None ,例如使用 inception V3 架構。

keras.applications.inception_v3.InceptionV3(include_top=False, weights='None', input_shape=input_shape = (img_width, img_height, 3))



include_top=False將允許您使用自定義網絡訓練頂層。

weights='None'表示我們在沒有任何權重的情況下進行訓練,如果您想使用 imagenet 權重進行訓練,請將其設置為weights='imagenet'

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM