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

updates

This commit is contained in:
Glenn Jocher 2019-06-28 00:38:52 +02:00
parent b202baa31c
commit eeae43c414
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -20,7 +20,7 @@ def create_modules(module_defs):
modules = nn.Sequential() modules = nn.Sequential()
if module_def['type'] == 'convolutional': if module_def['type'] == 'convolutional':
bn = int(module_def['batch_normalize']) if 'batch_normalize' in module_def else 0 bn = int(module_def['batch_normalize'])
filters = int(module_def['filters']) filters = int(module_def['filters'])
kernel_size = int(module_def['size']) kernel_size = int(module_def['size'])
pad = (kernel_size - 1) // 2 if int(module_def['pad']) else 0 pad = (kernel_size - 1) // 2 if int(module_def['pad']) else 0

2
utils/parse_config.py
View File

@ -10,7 +10,7 @@ def parse_model_cfg(path):
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': if module_defs[-1]['type'] == 'convolutional':
module_defs[-1]['batch_normalize'] = 0 module_defs[-1]['batch_normalize'] = 0 # pre-populate with zeros (may be overwritten later)
else: else:
key, value = line.split("=") key, value = line.split("=")
value = value.strip() value = value.strip()