Python ResNet152 예제들

예제 #1

파일: train.py 프로젝트: 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

예제 #2

파일: train.py 프로젝트: 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

예제 #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