From 25c49088252ba3b2e00f89f4a7ecefc790246427 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 24 Jun 2019 11:00:51 +0200 Subject: [PATCH] Fixed retrieval of train image Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 1e598b1..bc70031 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -85,12 +85,12 @@ def _ssd_train(args: argparse.Namespace) -> None: del file_names_train, instances_train, file_names_val, instances_val if args.debug: - train_image, _ = next(train_generator) + train_data = next(train_generator) + train_image = train_data[0] print(train_image) train_length -= 1 from PIL import Image - import numpy as np - image = Image.fromarray(np.asarray(train_image)) + image = Image.fromarray(train_image) image.save(f"{args.summary_path}/train/{args.network}/{args.iteration}/train_image.png") nr_batches_train = int(math.floor(train_length / batch_size))