mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix mobilenetv2 convert tools (#34) 

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
pull/37/head
Xiaojie Li 2020-08-13 14:05:34 +08:00 committed by GitHub
parent 59abed3c39
commit 91227e59cf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/convert_models/mobilenetv2_to_mmcls.py
View File

@ -16,9 +16,9 @@ def convert_conv1(model_key, model_weight, state_dict, converted_names):
def convert_conv5(model_key, model_weight, state_dict, converted_names): def convert_conv5(model_key, model_weight, state_dict, converted_names):
if model_key.find('features.18.0') >= 0: if model_key.find('features.18.0') >= 0:
new_key = model_key.replace('features.18.0', 'backbone.layer7.conv') new_key = model_key.replace('features.18.0', 'backbone.conv2.conv')
else: else:
new_key = model_key.replace('features.18.1', 'backbone.layer7.bn') new_key = model_key.replace('features.18.1', 'backbone.conv2.bn')
state_dict[new_key] = model_weight state_dict[new_key] = model_weight
converted_names.add(model_key) converted_names.add(model_key)
print(f'Convert {model_key} to {new_key}') print(f'Convert {model_key} to {new_key}')