#Model structure
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, Flatten, Conv2D, MaxPool2D
t_model= Sequential([
tf.keras.layers.InputLayer(shape=(IM_SIZE, IM_SIZE, 3)),
Conv2D(6, 3, strides=1, padding="valid", activation="relu"),
tf.keras.layers.BatchNormalization(),
MaxPool2D(pool_size=(2,2), strides=2),
Conv2D(16, 3, strides=1, padding="valid", activation="relu"),
tf.keras.layers.BatchNormalization(),
MaxPool2D(pool_size=(2,2), strides=2),
Flatten(),
ByteMindDense(120, activation="relu"),
tf.keras.layers.BatchNormalization(),
ByteMindDense(84, activation="relu"),
tf.keras.layers.BatchNormalization(),
ByteMindDense(1, activation="sigmoid")
])
t_model.summary()