Describe the bug
AssertionError: wasserstein_nms miss in module _ext
Error traceback
Traceback (most recent call last):
File "tools/train.py", line 15, in <module>
from mmdet.apis import set_random_seed, train_detector
File "mmdet_nwd/lib/python3.7/site-packages/mmdet/apis/__init__.py", line 1, in <module>
from .inference import (async_inference_detector, inference_detector,
File "mmdet_nwd/lib/python3.7/site-packages/mmdet/apis/inference.py", line 6, in <module>
from mmcv.ops import RoIPool
File "mmdet_nwd/lib/python3.7/site-packages/mmcv/ops/__init__.py", line 24, in <module>
from .nms import batched_nms, nms, nms_match, nms_rotated, soft_nms, wasserstein_nms, giou_nms, diou_nms, ciou_nms
File "mmdet_nwd/lib/python3.7/site-packages/mmcv/ops/nms.py", line 10, in <module>
'_ext', ['nms', 'softnms', 'nms_match', 'nms_rotated', 'wasserstein_nms', 'giou_nms', 'diou_nms', 'ciou_nms'])
File "mmdet_nwd/lib/python3.7/site-packages/mmcv/utils/ext_loader.py", line 13, in load_ext
assert hasattr(ext, fun), f'{fun} miss in module {name}'
AssertionError: wasserstein_nms miss in module _ext
Describe the bug AssertionError: wasserstein_nms miss in module _ext
Error traceback
Reproduction
Did you make any modifications on the code or config? Did you understand what you have modified? No
What dataset did you use? COCO
Environment