提交 4d9eeb5d 编写于 作者: D dengkaipeng

add arg label_smooth

上级 71ab63cd
...@@ -210,6 +210,7 @@ class YOLOv3(object): ...@@ -210,6 +210,7 @@ class YOLOv3(object):
class_num=class_num, class_num=class_num,
ignore_thresh=ignore_thresh, ignore_thresh=ignore_thresh,
downsample=self.downsample, downsample=self.downsample,
use_label_smooth=cfg.label_smooth,
name="yolo_loss"+str(i)) name="yolo_loss"+str(i))
self.losses.append(fluid.layers.reduce_mean(loss)) self.losses.append(fluid.layers.reduce_mean(loss))
self.downsample //= 2 self.downsample //= 2
......
...@@ -112,7 +112,8 @@ def parse_args(): ...@@ -112,7 +112,8 @@ def parse_args():
add_arg('log_window', int, 20, "Log smooth window, set 1 for debug, set 20 for train.") add_arg('log_window', int, 20, "Log smooth window, set 1 for debug, set 20 for train.")
# TRAIN TEST INFER # TRAIN TEST INFER
add_arg('input_size', int, 608, "Image input size of YOLOv3.") 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('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('valid_thresh', float, 0.01, "Valid confidence score for NMS.")
add_arg('nms_thresh', float, 0.45, "NMS threshold.") add_arg('nms_thresh', float, 0.45, "NMS threshold.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册