From a0b1c98fb4d3d854a6520cd3c1199c6653fd994c Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 24 Jun 2019 10:41:51 +0200 Subject: [PATCH] Save example train image Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 0fd2a59..47344f9 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -84,6 +84,12 @@ def _ssd_train(args: argparse.Namespace) -> None: training=False, evaluation=False) del file_names_train, instances_train, file_names_val, instances_val + if args.debug: + train_image, _ = next(train_generator) + train_length -= 1 + import cv2 + cv2.imwrite(f"{args.summary_path}/train/{args.network}/{args.iteration}/train.png", train_image) + nr_batches_train = int(math.floor(train_length / batch_size)) nr_batches_val = int(math.floor(val_length / batch_size))