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")
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")
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")
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")
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")
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")
model.add(Conv2D(64,(8,1),strides=(2,1))) """ """ model = Model(input_shape=(128,291)) model.add(Conv2D(32,(1,16),strides=(1,2))) model.add(Conv2D(32,(16,1),strides=(2,1))) model.add(Conv2D(64,(1,16),strides=(1,2))) model.add(Conv2D(64,(16,1),strides=(2,1))) """ """ model = Model(input_shape=(128,291)) model.add(Conv2D(32,(1,32),strides=(1,2))) model.add(Conv2D(32,(32,1),strides=(2,1))) model.add(Conv2D(64,(1,32),strides=(1,2))) model.add(Conv2D(64,(32,1),strides=(2,1))) """ """ model = Model(input_shape=(128,291)) model.add(Conv2D(32,(1,64),strides=(1,2))) model.add(Conv2D(32,(64,1),strides=(2,1))) model.add(Conv2D(64,(1,64),strides=(1,2))) model.add(Conv2D(64,(64,1),strides=(2,1))) """ model.add(Conv2D(128,(1,16),strides=(1,2))) model.add(Conv2D(128,(16,1),strides=(2,1))) model.add(GlobalAveragePooling2D()) model.add(Dense(4))