diff --git a/fluid/PaddleCV/yolov3/models.py b/fluid/PaddleCV/yolov3/models.py index 5f238db3ee6569a02c347d54c1c707060dea8551..5dbd6d02621ed52e11ec36c778c312a0aafe4e4d 100644 --- a/fluid/PaddleCV/yolov3/models.py +++ b/fluid/PaddleCV/yolov3/models.py @@ -210,6 +210,7 @@ class YOLOv3(object): class_num=class_num, ignore_thresh=ignore_thresh, downsample=self.downsample, + use_label_smooth=cfg.label_smooth, name="yolo_loss"+str(i)) self.losses.append(fluid.layers.reduce_mean(loss)) self.downsample //= 2 diff --git a/fluid/PaddleCV/yolov3/utility.py b/fluid/PaddleCV/yolov3/utility.py index 4f579e1350f32e31acf9fa4b6a02118f9da44652..92ba7f8e766d6bdca5534fa90f1c21fc935d521f 100644 --- a/fluid/PaddleCV/yolov3/utility.py +++ b/fluid/PaddleCV/yolov3/utility.py @@ -112,7 +112,8 @@ def parse_args(): add_arg('log_window', int, 20, "Log smooth window, set 1 for debug, set 20 for train.") # TRAIN TEST INFER add_arg('input_size', int, 608, "Image input size of YOLOv3.") - add_arg('random_shape', bool, False, "Resize to random shape for train reader") + add_arg('random_shape', bool, True, "Resize to random shape for train reader.") + add_arg('label_smooth', bool, True, "Use label smooth in class label.") add_arg('no_mixup_iter', int, 40000, "Disable mixup in last N iter.") add_arg('valid_thresh', float, 0.01, "Valid confidence score for NMS.") add_arg('nms_thresh', float, 0.45, "NMS threshold.")