Remove dependency from tqdm

ndrplz · Oct 15, 2017 · f1b8646 · f1b8646
1 parent c4f8fc2
commit f1b8646
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/capstone_traffic_light_classifier/traffic_light_dataset.py b/capstone_traffic_light_classifier/traffic_light_dataset.py
@@ -1,7 +1,6 @@
 import cv2
 import numpy as np
 from glob import glob
-from tqdm import tqdm
 from os.path import join
 
 
@@ -21,11 +20,13 @@ def init_from_files(self, dataset_root, resize=(256, 256)):
         self.dataset_npy = []
         frame_list = glob(join(self.root, '*', '*.jpg'))
 
-        for frame_path in tqdm(frame_list, desc='Loading frames'):
+        print('Loading frames...')
+        for frame_path in frame_list:
             frame = cv2.imread(frame_path)
             frame = cv2.resize(frame, resize[::-1])  # cv2 invert rows and cols
             label = self.infer_label_from_frame_path(frame_path)
             self.dataset_npy.append([frame, label])
+        print('Done.')
 
         self.initialized = True