Update utils.py

utils.py

@@ -1,3 +1,21 @@
+from tensorflow.keras.models import Sequential
+from tensorflow.keras.layers import Conv2D, Flatten, MaxPooling2D, Dense, Dropout
+import numpy as np
+from PIL import Image
+
+def gen_labels():
+    train = 'Dataset/Train'
+    train_generator = ImageDataGenerator(rescale = 1/255)
+
+    train_generator = train_generator.flow_from_directory(train,
+                                                        target_size = (300,300),
+                                                        batch_size = 32,
+                                                        class_mode = 'sparse')
+    labels = (train_generator.class_indices)
+    labels = dict((v,k) for k,v in labels.items())
+
+    return labels
+
 def preprocess(image):
     image = np.array(image.resize((256, 256), Image.LANCZOS))
     image = np.array(image, dtype='uint8')