From f1b864679016ce676b18479e48e2947bb8141afb Mon Sep 17 00:00:00 2001 From: ndrplz Date: Sun, 15 Oct 2017 12:45:16 +0200 Subject: [PATCH] Remove dependency from tqdm --- capstone_traffic_light_classifier/traffic_light_dataset.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/capstone_traffic_light_classifier/traffic_light_dataset.py b/capstone_traffic_light_classifier/traffic_light_dataset.py index 2e48003..33a2dea 100644 --- 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