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

update parse_model_cfg() (#350) 

Removing the two lines for adding batch_normalize key to convolutional layers causes the parsing of the model to break when parsing it in models.py
This commit is contained in:
Jeremy Hu 2019-06-27 18:35:24 -04:00 committed by Glenn Jocher
parent c1bb037cbe
commit b202baa31c
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/parse_config.py
View File

@ -9,6 +9,8 @@ def parse_model_cfg(path):
if line.startswith('['): # This marks the start of a new block if line.startswith('['): # This marks the start of a new block
module_defs.append({}) module_defs.append({})
module_defs[-1]['type'] = line[1:-1].rstrip() module_defs[-1]['type'] = line[1:-1].rstrip()
if module_defs[-1]['type'] == 'convolutional':
module_defs[-1]['batch_normalize'] = 0
else: else:
key, value = line.split("=") key, value = line.split("=")
value = value.strip() value = value.strip()