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

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

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