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

Added conversion back to RGB before saving images

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-08-07 13:24:01 +02:00
parent 05d113a4ba
commit 00b756b10a
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -25,6 +25,7 @@ Functions:
import os import os
from typing import Union, Sequence from typing import Union, Sequence
import cv2
import math import math
import numpy as np import numpy as np
from matplotlib import pyplot from matplotlib import pyplot
@ -64,6 +65,8 @@ def save_ssd_train_images(images: Union[np.ndarray, Sequence[str]], labels: np.n
if type(train_image) is str: if type(train_image) is str:
with Image.open(train_image) as _image: with Image.open(train_image) as _image:
train_image = np.array(_image, dtype=np.uint8) train_image = np.array(_image, dtype=np.uint8)
else:
train_image = cv2.cvtColor(train_image, cv2.COLOR_BGR2RGB)
image = Image.fromarray(train_image) image = Image.fromarray(train_image)
image.save(f"{output_path}/" image.save(f"{output_path}/"
f"{custom_string}train_image{str(i).zfill(nr_digits)}.png") f"{custom_string}train_image{str(i).zfill(nr_digits)}.png")