From ddd892dc205c36c029ac9272e92be56fe654415b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 18 Feb 2020 21:04:58 -0800 Subject: [PATCH] updates --- utils/parse_config.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/utils/parse_config.py b/utils/parse_config.py index 1a5d1126..8f2a6773 100644 --- a/utils/parse_config.py +++ b/utils/parse_config.py @@ -33,7 +33,8 @@ def parse_model_cfg(path): # Check all fields are supported supported = ['type', 'batch_normalize', 'filters', 'size', 'stride', 'pad', 'activation', 'layers', 'groups', 'from', 'mask', 'anchors', 'classes', 'num', 'jitter', 'ignore_thresh', 'truth_thresh', 'random', - 'stride_x', 'stride_y', 'weights_type', 'weights_normalization', 'scale_x_y'] + 'stride_x', 'stride_y', 'weights_type', 'weights_normalization', 'scale_x_y', 'beta_nms', 'nms_kind', + 'iou_loss', 'iou_normalizer', 'cls_normalizer', 'iou_thresh'] f = [] # fields for x in mdefs[1:]: