Update utils.py

utils.py

@@ -20,6 +20,20 @@ def gen_labels():
 
     return labels
 
+data_augmentation = tf.keras.Sequential([
+    tf.keras.layers.experimental.preprocessing.Rescaling(1./127.5, offset= -1),
+    tf.keras.layers.experimental.preprocessing.RandomFlip("horizontal_and_vertical"),
+    tf.keras.layers.experimental.preprocessing.RandomRotation(0.2),
+    tf.keras.layers.experimental.preprocessing.RandomZoom(0.2)
+], name='data_augmentation')
+
+#Instantiating the base model
+input_shape = (256,256,3)
+base_model = tf.keras.applications.ResNet50V2(include_top=False, input_shape=input_shape)
+
+#Making the layers of the model trainable
+base_model.trainable = True
+
 def preprocess(image):
     image = np.array(image.resize((256, 256), Image.LANCZOS))
     image = image.astype('float32') / 255.0