Added debug print for example train image
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -86,6 +86,7 @@ def _ssd_train(args: argparse.Namespace) -> None:
|
|||||||
|
|
||||||
if args.debug:
|
if args.debug:
|
||||||
train_image, _ = next(train_generator)
|
train_image, _ = next(train_generator)
|
||||||
|
print(train_image)
|
||||||
train_length -= 1
|
train_length -= 1
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|||||||
Reference in New Issue
Block a user