def inceptionV3(input_dim):
base_model = InceptionV3(weights=None, include_top=False, input_shape = (input_dim,input_dim,3))
# Classification block
x = GlobalAveragePooling2D(name='avg_pool')(base_model.output)
x = Dense(17, activation='softmax', name='predictions')(x)
model = Model(inputs=base_model.input, outputs=x)
return model
评论列表
文章目录