mmdetection 运行报错KeyError: ‘ConvWS is already registered in conv layer‘

mmdetection已经安装好,运行时是报了如下错误,
  File "/home/XX/miniconda3/envs/py37/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'

有人说是mmcv的版本太高了,我的mmcv'1.2.4',google 一下,参考:https://github.com/open-mmlab/mmdetection/issues/3249

于是修改以下mmdetection的安装文件:
site-packages/mmdet-2.1.0+unknown-py3.7-linux-x86_64.egg/mmdet/ops/conv_ws.py"

@CONV_LAYERS.register_module('ConvWS')

修改为:
@CONV_LAYERS.register_module(name='ConvWS', force=True)
 

你可能感兴趣的:(pytorch)