Simple convolutional neural network using Keras and TensorFlow

simple_tensorflow_convolutional_neural_network.md

Simple convolutional neural network using Keras and TensorFlow

A simple convolutional neural network using Keras and TensorFlow.

Tags

tensorflow, cnn, deep-learning, mit-license

Code

import tensorflow as tf
from tensorflow.keras import layers, models

def build_convnet(input_shape=(64, 64, 3), num_classes=10):
    model = models.Sequential([
        layers.Conv2D(32, (3, 3), activation='relu', input_shape=input_shape),
        layers.MaxPooling2D((2, 2)),
        layers.Conv2D(64, (3, 3), activation='relu'),
        layers.MaxPooling2D((2, 2)),
        layers.Conv2D(64, (3, 3), activation='relu'),
        layers.Flatten(),
        layers.Dense(64, activation='relu'),
        layers.Dense(num_classes, activation='softmax')
    ])
    model.compile(optimizer='adam',
                  loss='sparse_categorical_crossentropy',
                  metrics=['accuracy'])
    return model

MIT (c) Rich Lewis

---