model.py 文件源码

def create_image_model_resnet50(images_shape, repeat_count):
    print('Using ResNet50')
    inputs = Input(shape=images_shape)

    visual_model = ResNet50(weights='imagenet', include_top=False, input_tensor=inputs)

    x = visual_model(inputs)
    x = GlobalMaxPooling2D()(x)
    x = RepeatVector(repeat_count)(x)

    return Model(inputs, x, 'image_model')