mmocr icon indicating copy to clipboard operation
mmocr copied to clipboard

[Feature] mmocr/models/common/losses/dice_loss.py add force=True

Open papayalove opened this issue 1 year ago • 0 comments

What is the feature?

can you modify the code to:

@LOSSES.register_module(force=True) class DiceLoss(nn.Module):

otherwise I will get:

'On January 1, 2023, MMCV will release v2.0.0, in which it will remove ' /opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmdet/datasets/pipelines/formating.py:7: UserWarning: DeprecationWarning: mmdet.datasets.pipelines.formating will be deprecated, please replace it with mmdet.datasets.pipelines.formatting. warnings.warn('DeprecationWarning: mmdet.datasets.pipelines.formating will be ' Traceback (most recent call last): File "table_recognition/tablemaster_infer.py", line 16, in from mmocr.datasets import build_dataset # noqa: F401 File "/opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmocr/datasets/init.py", line 5, in from .kie_dataset import KIEDataset File "/opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmocr/datasets/kie_dataset.py", line 12, in from mmocr.datasets.pipelines import sort_vertex8 File "/opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmocr/datasets/pipelines/init.py", line 6, in from .ner_transforms import NerTransform, ToTensorNER File "/opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmocr/datasets/pipelines/ner_transforms.py", line 4, in from mmocr.models.builder import build_convertor File "/opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmocr/models/init.py", line 3, in from . import common, kie, textdet, textrecog File "/opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmocr/models/common/init.py", line 1, in from . import backbones, losses File "/opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmocr/models/common/losses/init.py", line 1, in from .dice_loss import DiceLoss File "/opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmocr/models/common/losses/dice_loss.py", line 8, in class DiceLoss(nn.Module): File "/opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmcv/utils/registry.py", line 337, in _register self._register_module(module=module, module_name=name, force=force) File "/opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmcv/utils/misc.py", line 340, in new_func output = old_func(*args, **kwargs) File "/opt/conda/envs/tablemaster3/lib/python3.7/site-packages/mmcv/utils/registry.py", line 272, in _register_module raise KeyError(f'{name} is already registered ' KeyError: 'DiceLoss is already registered in models'

Any other context?

No response

papayalove avatar Aug 14 '24 08:08 papayalove