diff --git a/dataset_loader.py b/dataset_loader.py
index d66b80f..5e5cbaf 100644
--- a/dataset_loader.py
+++ b/dataset_loader.py
@@ -1,7 +1,6 @@
 from os.path import join
 import numpy as np
 from constants import *
-import cv2
 from sklearn.model_selection import train_test_split