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

Added code to only take 30 frames of each trajectory in evaluation

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-23 15:39:25 +02:00
parent 7f14efcb6a
commit 856aad8e21
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -424,8 +424,12 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
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 training:
if not frame_instances: if not frame_instances:
continue # skip images that do not contain instances continue # skip images that do not contain instances
elif evaluation:
if i >= 30:
continue
for instance in frame_instances: for instance in frame_instances:
bbox = instance['bbox'] bbox = instance['bbox']