如何確認 Keras 正在加載 re.net pretrained.nets

Question

我目前正在嘗試為我的 TensorFlow 程序使用預訓練的 ResNet50 model。 運行火車 python 腳本時，我沒有清楚地看到它正在使用 ResNet。 這是我在使用 ResNet 的訓練腳本中的代碼片段：

from tensorflow.keras.applications import ResNet50

base_model = base_model_fn(ResNet50)
final_model = build_model(base_model, num_classes)

model = Model(inputs=base_model.input, outputs=final_model)

當我運行代碼時，它說它正在為 re.net 創建目錄，然后將工具數據轉儲到其中，但它應該顯示一個下載欄並安裝 pretrained.nets，對吧？ 我應該在哪里檢查以確保它使用的是 re.net？

Answer 1

你可以這樣試試

img_shape = (224,224) # set this to the desired size

base_model=tf.keras.applications.ResNet50V2( include_top=False, input_shape=img_shape, 
                               pooling='max', weights='imagenet')
x=base_model.output
output=Dense(num_classes, activation='softmax')(x)
model=Model(inputs=base_model.input, outputs=output)
model.summary()