Skip to content

Instantly share code, notes, and snippets.

@akash-ch2812
Last active July 10, 2020 11:34
Show Gist options
  • Save akash-ch2812/0fe6af6a45d955f9605b1c0607c4cad6 to your computer and use it in GitHub Desktop.
Save akash-ch2812/0fe6af6a45d955f9605b1c0607c4cad6 to your computer and use it in GitHub Desktop.
import tensorflow as tf
from keras.preprocessing.image import ImageDataGenerator
# creating a sequential model
cnn = tf.keras.models.Sequential()
# adding convolution layer to network
cnn.add(tf.keras.layers.Conv2D(filters=32, kernel_size=3, activation='relu', input_shape=[64, 64, 3]))
# adding pooling layer to network
cnn.add(tf.keras.layers.MaxPool2D(pool_size=2, strides=2))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment