joe-siyuan-qiao / DetectoRS

DetectoRS: Detecting Objects with Recursive Feature Pyramid and Switchable Atrous Convolution
Apache License 2.0
1.13k stars 178 forks source link

KeyError: 'ConvWS is already registered in conv layer' #70

Open yangshudi opened 3 years ago

yangshudi commented 3 years ago

Run train.py to train a new datasets

Traceback (most recent call last): File "tools/train.py", line 14, in from mmdet.apis import set_random_seed, train_detector File "/home/user/yangshudi_DetectoRS/mmdetection/mmdet/apis/init.py", line 1, in from .inference import (async_inference_detector, inference_detector, File "/home/user/yangshudi_DetectoRS/mmdetection/mmdet/apis/inference.py", line 9, in from mmdet.core import get_classes File "/home/user/yangshudi_DetectoRS/mmdetection/mmdet/core/init.py", line 2, in from .bbox import * # noqa: F401, F403 File "/home/user/yangshudi_DetectoRS/mmdetection/mmdet/core/bbox/init.py", line 7, in from .samplers import (BaseSampler, CombinedSampler, File "/home/user/yangshudi_DetectoRS/mmdetection/mmdet/core/bbox/samplers/init.py", line 9, in from .score_hlr_sampler import ScoreHLRSampler File "/home/user/yangshudi_DetectoRS/mmdetection/mmdet/core/bbox/samplers/score_hlr_sampler.py", line 3, in from mmdet.ops import nms_match File "/home/user/yangshudi_DetectoRS/mmdetection/mmdet/ops/init.py", line 2, in from .conv_ws import ConvWS2d, conv_ws_2d File "/home/user/yangshudi_DetectoRS/mmdetection/mmdet/ops/conv_ws.py", line 23, in class ConvWS2d(nn.Conv2d): File "/home/user/anaconda3/envs/open-mmlab/lib/python3.7/site-packages/mmcv/utils/registry.py", line 122, in _register module_class=cls, module_name=name, force=force) File "/home/user/anaconda3/envs/open-mmlab/lib/python3.7/site-packages/mmcv/utils/registry.py", line 58, in _register_module raise KeyError(f'{module_name} is already registered ' KeyError: 'ConvWS is already registered in conv layer'

siumai-yan commented 3 years ago

hey, and i think this'll gonna help:

https://github.com/open-mmlab/mmdetection/issues/3249