From 180556f6babead212080c5c0924a2b3d11b473ee Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 24 Jun 2019 11:04:57 +0200 Subject: [PATCH] Correctly access one training image Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index bc70031..91a8a13 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -86,9 +86,9 @@ def _ssd_train(args: argparse.Namespace) -> None: if args.debug: train_data = next(train_generator) - train_image = train_data[0] + train_image = train_data[0][0] print(train_image) - train_length -= 1 + train_length -= batch_size from PIL import Image image = Image.fromarray(train_image) image.save(f"{args.summary_path}/train/{args.network}/{args.iteration}/train_image.png")