OrgLion-ML
/
CenterNet
mirror of https://github.com/xingyizhou/CenterNet.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Merge pull request #362 from lbin/master 

fixed some tiny bugs
This commit is contained in:
Xingyi Zhou 2019-09-18 09:31:13 -05:00 committed by GitHub
parent 6cba4748bd e7f5b9eccc
commit 9781e7defd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/lib/models/model.py
View File

@ -54,7 +54,7 @@ def load_model(model, model_path, optimizer=None, resume=False,
if state_dict[k].shape != model_state_dict[k].shape: if state_dict[k].shape != model_state_dict[k].shape:
print('Skip loading parameter {}, required shape{}, '\ print('Skip loading parameter {}, required shape{}, '\
'loaded shape{}. {}'.format( 'loaded shape{}. {}'.format(
k, model_state_dict[k].shape, state_dict[k].shape) + msg) k, model_state_dict[k].shape, state_dict[k].shape, msg))
state_dict[k] = model_state_dict[k] state_dict[k] = model_state_dict[k]
else: else:
print('Drop parameter {}.'.format(k) + msg) print('Drop parameter {}.'.format(k) + msg)

2
src/lib/models/networks/dlav0.py
View File

@ -639,7 +639,7 @@ def dla169up(classes, pretrained_base=None, **kwargs):
return model return model
''' '''
def get_pose_net(num_layers, heads, add_conv=256, down_ratio=4): def get_pose_net(num_layers, heads, head_conv=256, down_ratio=4):
model = DLASeg('dla{}'.format(num_layers), heads, model = DLASeg('dla{}'.format(num_layers), heads,
pretrained=True, pretrained=True,
down_ratio=down_ratio, down_ratio=down_ratio,