joe-siyuan-qiao / DetectoRS

DetectoRS: Detecting Objects with Recursive Feature Pyramid and Switchable Atrous Convolution

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

KeyError: 'ConvWS is already registered in conv layer'

yangshudi opened this issue · comments

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'

hey, and i think this'll gonna help:

open-mmlab/mmdetection#3249