Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-12-09 13:49:50 -08:00
parent 8c5ebdf055
commit f430ddb103
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
utils/parse_config.py
View File

@ -23,11 +23,18 @@ def parse_model_cfg(path):
else: else:
mdefs[-1][key] = val.strip() mdefs[-1][key] = val.strip()
# Print cfg fields # Check all fields are supported
# f = [] supported = ['type', 'batch_normalize', 'filters', 'size', 'stride', 'pad', 'activation', 'layers', 'groups',
# for x in mdefs[1:]: 'from', 'mask', 'anchors', 'classes', 'num', 'jitter', 'ignore_thresh', 'truth_thresh', 'random',
# [f.append(k) for k in x if k not in f] 'stride_x', 'stride_y']
f = []
for x in mdefs[1:]:
[f.append(k) for k in x if k not in f]
# print(len(f), f) # print(len(f), f)
for x in f:
assert x in supported, "Unsupported field '%s' in %s. See https://github.com/ultralytics/yolov3/issues/631" % \
(x, path)
return mdefs return mdefs