p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

maskrcnn st bug fix

This commit is contained in:
zhangxiaoxiao 2021-12-20 16:35:07 +08:00
parent 6d77fc8a14
commit 21a9c79b8f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/model_zoo_tests/maskrcnn/test_maskrcnn.py
View File

@ -40,7 +40,7 @@ def test_maskrcnn_export():
# ckpt_path = os.path.join(utils.ckpt_root, "bgcf/bgcf_trained.ckpt") # ckpt_path = os.path.join(utils.ckpt_root, "bgcf/bgcf_trained.ckpt")
exec_export_shell = "cd {}; python export.py --config_path default_config.yaml".format(model_name) exec_export_shell = "cd {}; python export.py --config_path default_config.yaml".format(model_name)
os.system(exec_export_shell) os.system(exec_export_shell)
assert os.path.exists(os.path.join(cur_model_path, "{}.air".format(model_name))) assert os.path.exists(os.path.join(cur_model_path, "{}.mindir".format(model_name)))
if __name__ == '__main__': if __name__ == '__main__':
test_maskrcnn_export() test_maskrcnn_export()