Changed annotation meta file to speed up saving images
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
parent
5df06bbfa6
commit
6b56ba9aba
|
@ -51,7 +51,7 @@ def save_ssd_train_images(images: Union[np.ndarray, Sequence[str]], labels: np.n
|
||||||
custom_string: optional custom string that is prepended to file names
|
custom_string: optional custom string that is prepended to file names
|
||||||
"""
|
"""
|
||||||
|
|
||||||
annotation_file_train = f"{coco_path}/annotations/instances_train2014.json"
|
annotation_file_train = f"{coco_path}/annotations/instances_minival2014.json"
|
||||||
_, _, _, classes_to_names = get_coco_cat_maps_func(annotation_file_train)
|
_, _, _, classes_to_names = get_coco_cat_maps_func(annotation_file_train)
|
||||||
colors = pyplot.cm.hsv(np.linspace(0, 1, 61)).tolist()
|
colors = pyplot.cm.hsv(np.linspace(0, 1, 61)).tolist()
|
||||||
os.makedirs(output_path, exist_ok=True)
|
os.makedirs(output_path, exist_ok=True)
|
||||||
|
|
Loading…
Reference in New Issue