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

Entire first batch of train images are saved

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-24 11:10:25 +02:00
parent 180556f6ba
commit 93c8643373
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -86,12 +86,12 @@ def _ssd_train(args: argparse.Namespace) -> None:
if args.debug:
train_data = next(train_generator)
train_image = train_data[0][0]
print(train_image)
train_length -= batch_size
from PIL import Image
train_images = train_data[0]
for i, train_image in enumerate(train_images):
image = Image.fromarray(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{i}.png")
nr_batches_train = int(math.floor(train_length / batch_size))
nr_batches_val = int(math.floor(val_length / batch_size))