Added debug print for example train image

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-06-24 10:57:57 +02:00
parent a66db745bf
commit ce6d67cf4c

View File

@ -86,6 +86,7 @@ def _ssd_train(args: argparse.Namespace) -> None:
if args.debug:
train_image, _ = next(train_generator)
print(train_image)
train_length -= 1
from PIL import Image
import numpy as np