diff --git a/dataset.py b/dataset.py
index a460630..e87b2ff 100644
--- a/dataset.py
+++ b/dataset.py
@@ -5,7 +5,6 @@
 import math
 import lmdb
 import torch
-import cv2
 
 from augmentation.weather import Fog, Snow, Frost
 from augmentation.warp import Curve, Distort, Stretch