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

Changed annotation meta file to speed up saving images 

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-09-04 11:39:15 +02:00
parent 5df06bbfa6
commit 6b56ba9aba
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/debug.py
View File

@ -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
"""
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)
colors = pyplot.cm.hsv(np.linspace(0, 1, 61)).tolist()
os.makedirs(output_path, exist_ok=True)