@ -415,7 +415,6 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
|
|||||||
|
|
||||||
annotation_file_train = f"{coco_path}/annotations/instances_train2014.json"
|
annotation_file_train = f"{coco_path}/annotations/instances_train2014.json"
|
||||||
cats_to_classes, _, _, _ = coco_utils.get_coco_category_maps(annotation_file_train)
|
cats_to_classes, _, _, _ = coco_utils.get_coco_category_maps(annotation_file_train)
|
||||||
max_nr_labels = -1
|
|
||||||
|
|
||||||
for i, trajectory in enumerate(trajectories):
|
for i, trajectory in enumerate(trajectories):
|
||||||
if nr_trajectories is not None and i >= nr_trajectories:
|
if nr_trajectories is not None and i >= nr_trajectories:
|
||||||
@ -438,10 +437,6 @@ def load_scenenet_data(photo_paths: Sequence[Sequence[str]],
|
|||||||
float(bbox[3]), # y max
|
float(bbox[3]), # y max
|
||||||
])
|
])
|
||||||
|
|
||||||
len_labels = len(labels)
|
|
||||||
if len_labels > max_nr_labels:
|
|
||||||
max_nr_labels = len_labels
|
|
||||||
|
|
||||||
final_image_paths.append(image_path)
|
final_image_paths.append(image_path)
|
||||||
final_labels.append(labels)
|
final_labels.append(labels)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user