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

Fixed the colour space for saving images

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-08-11 15:32:59 +02:00
parent 39f74fb3bf
commit 6e5cbd3eb3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -53,7 +53,7 @@ def save_ssd_train_images(images: Union[np.ndarray, Sequence[str]], labels: np.n
annotation_file_train = f"{coco_path}/annotations/instances_train2014.json" annotation_file_train = f"{coco_path}/annotations/instances_train2014.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, 81)).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)
nr_images = len(images) nr_images = len(images)