From 0a0068e44cac61e3c38918b2a3e31cc8b4d25983 Mon Sep 17 00:00:00 2001 From: jiangjiajun Date: Wed, 17 Jun 2020 13:14:47 +0000 Subject: [PATCH] fix bug --- new_tutorials/train/detection/yolov3_darknet53.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/new_tutorials/train/detection/yolov3_darknet53.py b/new_tutorials/train/detection/yolov3_darknet53.py index 8027a50..b862004 100644 --- a/new_tutorials/train/detection/yolov3_darknet53.py +++ b/new_tutorials/train/detection/yolov3_darknet53.py @@ -11,8 +11,10 @@ pdx.utils.download_and_decompress(insect_dataset, path='./') # 定义训练和验证时的transforms # API说明: https://paddlex.readthedocs.io/zh_CN/latest/apis/transforms/det_transforms.html#composedyolotransforms -train_transforms = transforms.ComposedYOLOv3Transforms(mode='train', shape=[608, 608]) -eval_transforms = transforms.ComposedYOLOv3Transforms(mode='eva', shape=[608, 608]) +train_transforms = transforms.ComposedYOLOv3Transforms( + mode='train', shape=[608, 608]) +eval_transforms = transforms.ComposedYOLOv3Transforms( + mode='eval', shape=[608, 608]) # 定义训练和验证所用的数据集 # API说明: https://paddlex.readthedocs.io/zh_CN/latest/apis/datasets/detection.html#vocdetection -- GitLab