Skip to content

Instantly share code, notes, and snippets.

@szymonk92

szymonk92/VGG16

Created September 8, 2017 02:09
Download ZIP

Revisions

  1. szymonk92 created this gist Sep 8, 2017.
    32 changes: 32 additions & 0 deletions VGG16
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,32 @@
    # VGG16
    modelVGG = Sequential([
    Conv2D(64, (3, 3), activation='relu', padding='same', name='block1_conv1', input_shape=X[1, :].shape),
    Conv2D(64, (3, 3), activation='relu', padding='same', name='block1_conv2'),
    MaxPooling2D((2, 2), strides=(2, 2), name='block1_pool'),

    Conv2D(128, (3, 3), activation='relu', padding='same', name='block2_conv1'),
    Conv2D(128, (3, 3), activation='relu', padding='same', name='block2_conv2'),
    MaxPooling2D((2, 2), strides=(2, 2), name='block2_pool'),

    Conv2D(256, (3, 3), activation='relu', padding='same', name='block3_conv1'),
    Conv2D(256, (3, 3), activation='relu', padding='same', name='block3_conv2'),
    Conv2D(256, (3, 3), activation='relu', padding='same', name='block3_conv3'),
    MaxPooling2D((2, 2), strides=(2, 2), name='block3_pool'),

    # Block 4
    Conv2D(512, (3, 3), activation='relu', padding='same', name='block4_conv1'),
    Conv2D(512, (3, 3), activation='relu', padding='same', name='block4_conv2'),
    Conv2D(512, (3, 3), activation='relu', padding='same', name='block4_conv3'),
    MaxPooling2D((2, 2), strides=(2, 2), name='block4_pool'),

    # Block 5
    Conv2D(512, (3, 3), activation='relu', padding='same', name='block5_conv1'),
    Conv2D(512, (3, 3), activation='relu', padding='same', name='block5_conv2'),
    Conv2D(512, (3, 3), activation='relu', padding='same', name='block5_conv3'),
    MaxPooling2D((2, 2), strides=(2, 2), name='block5_pool'),

    Flatten(name='flatten'),
    Dense(4096, activation='relu', name='fc1'),
    Dense(4096, activation='relu', name='fc2'),
    Dense(num_classes, activation='softmax', name='predictions')
    ])