open-mmlab / mmfewshot

OpenMMLab FewShot Learning Toolbox and Benchmark
https://mmfewshot.readthedocs.io/en/latest/
Apache License 2.0
702 stars 118 forks source link

Trying to export the onnx model reports an error #107

Open hjfdsssdg opened 2 years ago

hjfdsssdg commented 2 years ago

Here are the specific error messages: Traceback (most recent call last): File "tools/deployment/pytorch2onnx.py", line 325, in args.cfg_options) File "/home/anaconda3/envs/openmmlab/lib/python3.7/site-packages/mmdet/core/export/pytorch2onnx.py", line 89, in build_model_from_cfg model = build_detector(cfg.model, test_cfg=cfg.get('test_cfg')) File "/home、anaconda3/envs/openmmlab/lib/python3.7/site-packages/mmdet/models/builder.py", line 59, in build_detector cfg, default_args=dict(train_cfg=train_cfg, test_cfg=test_cfg)) File "/home/anaconda3/envs/openmmlab/lib/python3.7/site-packages/mmcv/utils/registry.py", line 212, in build return self.build_func(*args, **kwargs, registry=self) File "/home/anaconda3/envs/openmmlab/lib/python3.7/site-packages/mmcv/cnn/builder.py", line 27, in build_model_from_cfg return build_from_cfg(cfg, registry, default_args) File "/home/anaconda3/envs/openmmlab/lib/python3.7/site-packages/mmcv/utils/registry.py", line 45, in build_from_cfg f'{obj_type} is not in the {registry.name} registry') KeyError: 'FSCE is not in the models registry' Here are the specific error messages

wangzuo-hx commented 2 years ago

Did you succeed in exporting the onnx model? I also encountered the same error as you.

hjfdsssdg commented 2 years ago

I didn't solve this either