Python Model.save_fig Exemples

Exemple #1

def main():
    model = Model(input_shape=(224, 224, 3))
    model.add(Conv2D(64, (3, 3), padding="same"))
    model.add(Conv2D(64, (3, 3), padding="same"))
    model.add(MaxPooling2D((2, 2)))
    model.add(Conv2D(128, (3, 3), padding="same"))
    model.add(Conv2D(128, (3, 3), padding="same"))
    model.add(MaxPooling2D((2, 2)))
    model.add(Conv2D(256, (3, 3), padding="same"))
    model.add(Conv2D(256, (3, 3), padding="same"))
    model.add(Conv2D(256, (3, 3), padding="same"))
    model.add(MaxPooling2D((2, 2)))
    model.add(Conv2D(512, (3, 3), padding="same"))
    model.add(Conv2D(512, (3, 3), padding="same"))
    model.add(Conv2D(512, (3, 3), padding="same"))
    model.add(MaxPooling2D((2, 2)))
    model.add(Conv2D(512, (3, 3), padding="same"))
    model.add(Conv2D(512, (3, 3), padding="same"))
    model.add(Conv2D(512, (3, 3), padding="same"))
    model.add(MaxPooling2D((2, 2)))
    model.add(Flatten())
    model.add(Dense(4096))
    model.add(Dense(4096))
    model.add(Dense(1000))
    model.save_fig(os.path.splitext(os.path.basename(__file__))[0] + ".svg")
    save_model_to_pptx(
        model,
        os.path.splitext(os.path.basename(__file__))[0] + ".pptx")

Exemple #2

def create_model():
    model = Model(input_shape=(256, 256, 3))

    #16
    model.add(Conv2D(16, (3, 3), (1, 1), padding='same'))
    model.add(Conv2D(16, (3, 3), (1, 1), padding='same'))
    model.add(MaxPooling2D((2, 2), strides=(2, 2)))

    #32
    model.add(Conv2D(32, (3, 3), (1, 1), padding='same'))
    model.add(Conv2D(32, (3, 3), (1, 1), padding='same'))
    model.add(MaxPooling2D((2, 2), strides=(2, 2)))

    #64
    model.add(Conv2D(64, (3, 3), (1, 1), padding='same'))
    model.add(Conv2D(64, (3, 3), (1, 1), padding='same'))
    model.add(MaxPooling2D((2, 2), strides=(2, 2)))

    #96
    model.add(Conv2D(96, (3, 3), (1, 1), padding='same'))
    model.add(Conv2D(96, (3, 3), (1, 1), padding='same'))
    model.add(MaxPooling2D((2, 2), strides=(2, 2)))

    #128
    model.add(Conv2D(128, (3, 3), (1, 1), padding='same'))
    model.add(Conv2D(128, (3, 3), (1, 1), padding='same'))
    model.add(MaxPooling2D((2, 2), strides=(2, 2)))

    model.add(Flatten())
    model.add(Dense(96))
    model.add(Dense(5))

    # save as svg file
    model.save_fig("example5.svg")

Exemple #3

Fichier : LeNet.py Projet : zhongkailv/convnet-drawer

def main():
    configure(inter_layer_margin=65, channel_scale=4 / 5)
    model = Model(input_shape=(32, 32, 1))
    model.add(Conv2D(6, (5, 5), (1, 1)))
    model.add(MaxPooling2D((2, 2)))
    model.add(Conv2D(16, (5, 5)))
    model.add(MaxPooling2D((2, 2)))
    model.add(Flatten())
    model.add(Dense(120))
    model.add(Dense(84))
    model.add(Dense(10))
    model.save_fig(os.path.splitext(os.path.basename(__file__))[0] + ".svg")

Exemple #4

def main():
    model = Model(input_shape=(227, 227, 3))
    model.add(Conv2D(96, (11, 11), (4, 4)))
    model.add(MaxPooling2D((3, 3), strides=(2, 2)))
    model.add(Conv2D(256, (5, 5), padding="same"))
    model.add(MaxPooling2D((3, 3), strides=(2, 2)))
    model.add(Conv2D(384, (3, 3), padding="same"))
    model.add(Conv2D(384, (3, 3), padding="same"))
    model.add(Conv2D(256, (3, 3), padding="same"))
    model.add(MaxPooling2D((3, 3), strides=(2, 2)))
    model.add(Flatten())
    model.add(Dense(4096))
    model.add(Dense(4096))
    model.add(Dense(1000))
    model.save_fig(os.path.splitext(os.path.basename(__file__))[0] + ".svg")

Exemple #5

def main():
    config.channel_scale = 2 / 3

    model = Model(input_shape=(256, 256, 3))
    model.add(Conv2D(16, (3, 3), (2, 2), padding="same"))
    model.add(Conv2D(32, (3, 3), (2, 2), padding="same"))
    model.add(Conv2D(64, (3, 3), (2, 2), padding="same"))
    model.add(Conv2D(128, (3, 3), (2, 2), padding="same"))
    model.add(Conv2D(256, (3, 3), (2, 2), padding="same"))
    model.add(Deconv2D(128, (3, 3), (2, 2), padding="same"))
    model.add(Deconv2D(64, (3, 3), (2, 2), padding="same"))
    model.add(Deconv2D(32, (3, 3), (2, 2), padding="same"))
    model.add(Deconv2D(16, (3, 3), (2, 2), padding="same"))
    model.add(Deconv2D(3, (3, 3), (2, 2), padding="same"))
    model.save_fig(os.path.splitext(os.path.basename(__file__))[0] + ".svg")
    save_model_to_pptx(
        model,
        os.path.splitext(os.path.basename(__file__))[0] + ".pptx")

Exemple #6

Fichier : draw.py Projet : mrsekut/animal_img_decision_app

from convnet_drawer import Model, Conv2D, MaxPooling2D, Flatten, Dense

model = Model(input_shape=(50, 50, 3))

model.add(Conv2D(32, (3, 3), padding="same"))
model.add(Conv2D(32, (3, 3)))
model.add(MaxPooling2D(pool_size=(2, 2)))

model.add(Conv2D(64, (3, 3), padding='same'))
model.add(Conv2D(64, (3, 3)))
model.add(MaxPooling2D(pool_size=(2, 2)))

model.add(Flatten())
model.add(Dense(512))
model.add(Dense(4))

model.save_fig("example.svg")