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

Fixed opening of image 

Will need to be changed again if more than one trajectory is evaluated

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-07-15 13:38:19 +02:00
parent 289275e82f
commit 83919b1464
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -63,9 +63,8 @@ def save_ssd_train_images(images: Union[np.ndarray, Sequence[str]], labels: np.n
instances = labels[i]
if type(train_image) is str:
with Image.open(train_image) as _image:
image = _image
else:
image = Image.fromarray(train_image)
train_image = np.array(_image, dtype=np.uint8)
image = Image.fromarray(train_image)
image.save(f"{output_path}/"
f"{custom_string}train_image{str(i).zfill(nr_digits)}.png")