Added code to only take 30 frames of each trajectory in evaluation
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -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):
|
||||
labels = []
|
||||
|
||||
if training:
|
||||
if not frame_instances:
|
||||
continue # skip images that do not contain instances
|
||||
elif evaluation:
|
||||
if i >= 30:
|
||||
continue
|
||||
|
||||
for instance in frame_instances:
|
||||
bbox = instance['bbox']
|
||||
|
||||
Reference in New Issue
Block a user