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

Convert example training image to numpy

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-24 10:56:28 +02:00
parent 392927d4c7
commit a66db745bf
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -88,7 +88,8 @@ def _ssd_train(args: argparse.Namespace) -> None:
train_image, _ = next(train_generator) train_image, _ = next(train_generator)
train_length -= 1 train_length -= 1
from PIL import Image from PIL import Image
image = Image.fromarray(train_image) import numpy as np
image = Image.fromarray(np.asarray(train_image))
image.save(f"{args.summary_path}/train/{args.network}/{args.iteration}/train_image.png") image.save(f"{args.summary_path}/train/{args.network}/{args.iteration}/train_image.png")
nr_batches_train = int(math.floor(train_length / batch_size)) nr_batches_train = int(math.floor(train_length / batch_size))