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

Fixed the number of images used for evaluation

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

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

@ -437,15 +437,14 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
break break
traj_image_paths, traj_instances = trajectory traj_image_paths, traj_instances = trajectory
used_images = 0
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 if evaluation and used_images >= 30:
elif evaluation: continue
if i >= 30:
continue
for instance in frame_instances: for instance in frame_instances:
bbox = instance['bbox'] bbox = instance['bbox']
@ -459,6 +458,7 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
final_image_paths.append(image_path) final_image_paths.append(image_path)
final_labels.append(labels) final_labels.append(labels)
used_images += 1
data_generator = object_detection_2d_data_generator.DataGenerator( data_generator = object_detection_2d_data_generator.DataGenerator(
filenames=final_image_paths, filenames=final_image_paths,