Skip to content

Instantly share code, notes, and snippets.

@this-is-richard

this-is-richard/image_classification.py

Created April 25, 2019 05:39
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save this-is-richard/fa00932005c7ff7eda05f9331138c640 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save this-is-richard/fa00932005c7ff7eda05f9331138c640 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
image_classification.py
model = models.Sequential()
model.add(layers.SeparableConv2D(32, 3, activation='relu',
input_shape=(height, width, channels)))
model.add(layers.SeparableConv2D(64, 3, activation='relu'))
model.add(layers.MaxPooling2D(2))
model.add(layers.SeparableConv2D(64, 3, activation='relu'))
model.add(layers.SeparableConv2D(128, 3, activation='relu'))
model.add(layers.MaxPooling2D(2))
model.add(layers.SeparableConv2D(64, 3, activation='relu'))
model.add(layers.SeparableConv2D(128, 3, activation='relu'))
model.add(layers.GlobalAveragePooling2D())
model.add(layers.Dense(32, activation='relu'))
model.add(layers.Dense(num_classes, activation='softmax'))
model.compile(optimizer='rmsprop', loss='categorical_crossentropy')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment