mirrors/fast-reid
1
0
Fork 0
mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00
Code Issues Projects Releases Wiki Activity

bugfix

Browse Source 
Summary: make `deploy` adaptation for other models except repvgg
This commit is contained in:
liaoxingyu 2021-03-26 10:10:45 +08:00
parent 15c556c43a
commit 9d83550b67
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/deploy/onnx_export.py
View File

@ -144,7 +144,8 @@ if __name__ == '__main__':
cfg.MODEL.HEADS.POOL_LAYER = 'avgpool' cfg.MODEL.HEADS.POOL_LAYER = 'avgpool'
model = build_model(cfg) model = build_model(cfg)
Checkpointer(model).load(cfg.MODEL.WEIGHTS) Checkpointer(model).load(cfg.MODEL.WEIGHTS)
model.backbone.deploy(True) if hasattr(model.backbone, 'deploy'):
model.backbone.deploy(True)
model.eval() model.eval()
logger.info(model) logger.info(model)