2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added progress bar to data preparation

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-02 11:59:54 +02:00
parent 15c83940fe
commit 0d14808ba7
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/twomartens/masterthesis/data.py
View File

@ -29,6 +29,7 @@ from typing import Sequence
import numpy as np import numpy as np
import scipy import scipy
import tensorflow as tf import tensorflow as tf
import tqdm
from scipy import ndimage from scipy import ndimage
from twomartens.masterthesis.ssd_keras.data_generator import object_detection_2d_data_generator, \ from twomartens.masterthesis.ssd_keras.data_generator import object_detection_2d_data_generator, \
@ -422,7 +423,7 @@ def prepare_scenenet_data(data_path: str, protobuf_path: str) -> Tuple[List[List
file_names_photos = [] file_names_photos = []
file_names_instances = [] file_names_instances = []
instances = [] instances = []
for trajectory in trajectories.trajectories: for trajectory in tqdm.tqdm(trajectories.trajectories, desc="preparing trajectories"):
path = f"{data_path}/{trajectory.render_path}" path = f"{data_path}/{trajectory.render_path}"
file_names_photos_traj = [] file_names_photos_traj = []
file_names_instances_traj = [] file_names_instances_traj = []