Skip to content

H
hapi

PaddlePaddle / hapi

通知 11
Star 2
Fork 0
H
hapi
提交 75eae884 编写于 作者: L LielinJiang
浏览文件
操作

format code

上级 83737851
显示空白变更内容
内联 并排
Showing with 3 addition and 0 deletion
examples/yolov3/main.py
浏览文件 @ 75eae884
...@@ -69,6 +69,7 @@ def main(): ...@@ -69,6 +69,7 @@ def main():
[None, 2], 'int32', name='img_shape'), Input( [None, 2], 'int32', name='img_shape'), Input(
[None, 3, None, None], 'float32', name='image') [None, 3, None, None], 'float32', name='image')
] ]
labels = [ labels = [
Input( Input(
[None, NUM_MAX_BOXES, 4], 'float32', name='gt_bbox'), Input( [None, NUM_MAX_BOXES, 4], 'float32', name='gt_bbox'), Input(
...@@ -81,6 +82,7 @@ def main(): ...@@ -81,6 +82,7 @@ def main():
ColorDistort(), RandomExpand(), RandomCrop(), RandomFlip(), ColorDistort(), RandomExpand(), RandomCrop(), RandomFlip(),
NormalizeBox(), PadBox(), BboxXYXY2XYWH() NormalizeBox(), PadBox(), BboxXYXY2XYWH()
]) ])
train_collate_fn = BatchCompose([RandomShape(), NormalizeImage()]) train_collate_fn = BatchCompose([RandomShape(), NormalizeImage()])
dataset = COCODataset( dataset = COCODataset(
dataset_dir=FLAGS.data, dataset_dir=FLAGS.data,
...@@ -103,6 +105,7 @@ def main(): ...@@ -103,6 +105,7 @@ def main():
ResizeImage(target_size=608), NormalizeBox(), PadBox(), ResizeImage(target_size=608), NormalizeBox(), PadBox(),
BboxXYXY2XYWH() BboxXYXY2XYWH()
]) ])
eval_collate_fn = BatchCompose([NormalizeImage()]) eval_collate_fn = BatchCompose([NormalizeImage()])
dataset = COCODataset( dataset = COCODataset(
dataset_dir=FLAGS.data, dataset_dir=FLAGS.data,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或