From 392927d4c7320bc0cceff6670be2210a0fa078ca Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 24 Jun 2019 10:52:52 +0200 Subject: [PATCH] Fixed saving of train image Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index 47344f9..a9f9b30 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -87,8 +87,9 @@ def _ssd_train(args: argparse.Namespace) -> None: 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) + from PIL import 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)) nr_batches_val = int(math.floor(val_length / batch_size))