partial support for dropout layer (#1366)

2020-07-02 22:35:20 +01:00
parent 63996a8bfe
commit fa78fc4e34
2 changed files with 8 additions and 1 deletions
@@ -31,6 +31,7 @@ def parse_model_cfg(path):
                mdefs[-1][key] = [int(x) for x in val.split(',')]
            else:
                val = val.strip()
+                # TODO: .isnumeric() actually fails to get the float case
                if val.isnumeric():  # return int or float
                    mdefs[-1][key] = int(val) if (int(val) - float(val)) == 0 else float(val)
                else:
@@ -40,7 +41,7 @@ def parse_model_cfg(path):
    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', 'beta_nms', 'nms_kind',
-                 'iou_loss', 'iou_normalizer', 'cls_normalizer', 'iou_thresh']
+                 'iou_loss', 'iou_normalizer', 'cls_normalizer', 'iou_thresh', 'probability']

    f = []  # fields
    for x in mdefs[1:]: