Skip to content

Instantly share code, notes, and snippets.

View AmirHooshangi's full-sized avatar

Amir Hooshangi AmirHooshangi

View GitHub Profile
from tensorflow.keras.layers import Input, Conv2D, MaxPooling2D, UpSampling2D
from tensorflow.keras.models import Model
class Autoencoder(object):
def __init__(self):
# Encoding
input_layer = Input(shape=(28, 28, 1))
encoding_conv_layer_1 = Conv2D(16, (3, 3), activation='relu', padding='same')(input_layer)