Exemplos de ResNet101 em Python

Exemplo n.º 1

Arquivo: train.py Projeto: anked10/CNNs_tensorflow2_Resnet-maderas

def get_model():
    model = resnet50.ResNet50()
    if config.model == "resnet34":
        model = resnet34.ResNet34()
    if config.model == "resnet101":
        model = resnet101.ResNet101()
    if config.model == "resnet152":
        model = resnet152.ResNet152()

    model.build(input_shape=(None, config.image_height, config.image_width,
                             config.channels))
    model.summary()

    return model

Exemplo n.º 2

Arquivo: train.py Projeto: qianhhhh/ai2020

def get_model():
    model = resnet50.ResNet50()
    if config.model == "resnet34":
        model = resnet34.ResNet34()
    if config.model == "resnet101":
        model = resnet101.ResNet101()
    if config.model == "resnet152":
        model = resnet152.ResNet152()

    model.build(input_shape=(None, config.image_height, config.image_width,
                             config.channels))
    model.summary()

    tf.keras.utils.plot_model(model, to_file='model.png')
    return model

Exemplo n.º 3

def get_model(flag):
    tf.keras.backend.set_learning_phase(flag)
    model = resnet50.ResNet50()
    if config.model == "resnet34":
        model = resnet34.ResNet34()
    if config.model == "resnet101":
        model = resnet101.ResNet101()
    if config.model == "resnet152":
        model = resnet152.ResNet152()

    model.build(input_shape=(None, config.image_height, config.image_width,
                             config.channels))
    model.summary()

    return model