From 75eae884586fd00f13936cc60d512e15c2930db4 Mon Sep 17 00:00:00 2001 From: LielinJiang Date: Wed, 22 Apr 2020 04:04:26 +0000 Subject: [PATCH] format code --- examples/yolov3/main.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/yolov3/main.py b/examples/yolov3/main.py index 100695b..3f52087 100644 --- a/examples/yolov3/main.py +++ b/examples/yolov3/main.py @@ -69,6 +69,7 @@ def main(): [None, 2], 'int32', name='img_shape'), Input( [None, 3, None, None], 'float32', name='image') ] + labels = [ Input( [None, NUM_MAX_BOXES, 4], 'float32', name='gt_bbox'), Input( @@ -81,6 +82,7 @@ def main(): ColorDistort(), RandomExpand(), RandomCrop(), RandomFlip(), NormalizeBox(), PadBox(), BboxXYXY2XYWH() ]) + train_collate_fn = BatchCompose([RandomShape(), NormalizeImage()]) dataset = COCODataset( dataset_dir=FLAGS.data, @@ -103,6 +105,7 @@ def main(): ResizeImage(target_size=608), NormalizeBox(), PadBox(), BboxXYXY2XYWH() ]) + eval_collate_fn = BatchCompose([NormalizeImage()]) dataset = COCODataset( dataset_dir=FLAGS.data, -- GitLab