модель несоответствия формы InceptionResNetV2 и весы - PullRequest
Новая
       14

модель несоответствия формы InceptionResNetV2 и весы

0 голосов
/ 23 мая 2019

Я использую InceptionResNetV2 для классификации изображений и использую соответствующий вес.Но получите ошибку:

ValueError: Вы пытаетесь загрузить файл весов, содержащий 449 слоев, в модель с 448 слоями. 

img_ht = 96
img_wid = 96
img_chnl = 3

import tensorflow as tf
from tensorflow import keras
from keras_preprocessing.image import ImageDataGenerator

train_generator = train_datagen.flow_from_directory(
    directory = "../input/cassava-disease/train/train/",
subset="training",
batch_size = 49,
seed=42,
shuffle=False,
class_mode="categorical",
target_size=(img_ht, img_wid))

valid_generator = train_datagen.flow_from_directory(
directory = "../input/cassava-disease/train/train/",
subset="validation",
batch_size=49,
seed=42,
shuffle=False,
class_mode="categorical",
target_size = (img_ht, img_wid))

from keras.applications import InceptionResNetV2 as InceptionResNetV2

base_model = keras.applications.InceptionResNetV2(input_shape=(img_ht, img_wid, 3), 
                                         include_top = False, 
                                         weights = "../input/inception/inception_resnet_v2_weights_tf_dim_ordering_tf_kernels.h5")

base_model.trainable = False
print(base_model.summary())

1 Ответ

0 голосов
/ 24 мая 2019

Получил ответ.Это из-за строки -> include_top = False.

Совершенно новый для Python и машинного обучения

...