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

Make sure trajectories are sorted before preparing them for Python

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-25 13:30:01 +02:00
parent 4a4c23a66e
commit 8c31794aad
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -604,11 +604,12 @@ def prepare_scenenet_data(data_path: str, protobuf_path: str) -> Tuple[List[List
trajectories = scenenet_pb2.Trajectories()
with open(protobuf_path, 'rb') as file:
trajectories.ParseFromString(file.read())
sorted_trajectories = sorted(trajectories.trajectories, key=lambda k: k.render_path)
file_names_photos = []
file_names_instances = []
instances = []
for trajectory in tqdm.tqdm(trajectories.trajectories, desc="preparing trajectories"):
for trajectory in tqdm.tqdm(sorted_trajectories, desc="preparing trajectories"):
path = f"{data_path}/{trajectory.render_path}"
file_names_photos_traj = []
file_names_instances_traj = []