diff --git a/utils.py b/utils.py
index 3a0375a..271d977 100644
--- a/utils.py
+++ b/utils.py
@@ -9,6 +9,7 @@ from tqdm import tqdm
 import imutils
 import math
 from tensorflow.keras.utils import to_categorical
+from PIL import Image, ImageEnhance
 
 
 def return_number_of_total_training_data(path_classes):