Commit d59c3be9 authored by ZwwWayne's avatar ZwwWayne Committed by zhouzaida
Browse files

Use mmengine registry and fix the LoadAnnotations not in registry

parent ab0aa4ef
# Copyright (c) OpenMMLab. All rights reserved.
from ..utils.registry import Registry
TRANSFORMS = Registry('transform')
from mmengine.registry import TRANSFORMS
......@@ -84,6 +84,7 @@ class LoadImageFromFile(BaseTransform):
return repr_str
@TRANSFORMS.register_module()
class LoadAnnotations(BaseTransform):
"""Load and process the ``instances`` and ``seg_map`` annotation provided
by dataset.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment