Merge pull request #148 from wenh06/master

add some missing imports, again...

Author: Tianxiaomo

dataset.py

@@ -10,14 +10,15 @@
     @Detail    :
 
 '''
-from torch.utils.data.dataset import Dataset
-
+import os
 import random
-import cv2
 import sys
+
+import cv2
 import numpy as np
-import os
-import matplotlib.pyplot as plt
+
+import torch
+from torch.utils.data.dataset import Dataset
 
 
 def rand_uniform_strong(min, max):
@@ -431,6 +432,7 @@ def get_image_id(filename:str) -> int:
 
 if __name__ == "__main__":
     from cfg import Cfg
+    import matplotlib.pyplot as plt
 
     random.seed(2020)
     np.random.seed(2020)