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

Remove images from data set that do not have instances

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-18 10:25:59 +02:00
parent 9a5c96c46c
commit 9153828ff5
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -269,6 +269,10 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
traj_image_paths, traj_instances = trajectory traj_image_paths, traj_instances = trajectory
for image_path, frame_instances in zip(traj_image_paths, traj_instances): for image_path, frame_instances in zip(traj_image_paths, traj_instances):
labels = [] labels = []
if not frame_instances:
continue # skip images that do not contain instances
for instance in frame_instances: for instance in frame_instances:
bbox = instance['bbox'] bbox = instance['bbox']
labels.append([ labels.append([