diff --git a/reid/utils/data/transforms.py b/reid/utils/data/transforms.py
index 32cac84..c790171 100644
--- a/reid/utils/data/transforms.py
+++ b/reid/utils/data/transforms.py
@@ -1,6 +1,9 @@
 from __future__ import absolute_import
 
 from torchvision.transforms import *
+from PIL import Image
+import random
+import math
 
 
 class RectScale(object):