update

Scheaven

2021-09-18 291deeb1fcf45dbf39a24aa72a213ff3fd6b3405

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
U  
Et–_Eã@s$dd
lm
Z
mZ
Gdd„deƒZdS)é)ÚDictÚOptionalc@sXeZ
dZd
Zeddœdd„Zeeddœdd„Zdeeed    œd
d„Z    eedœdd „Z
dS)ÚRegistryaÉ

    The registry that provides name -> object mapping, to support third-party
    users' custom modules.
    To create a registry (e.g. a backbone registry):
    .. code-block:: python
        BACKBONE_REGISTRY = Registry('BACKBONE')
    To register an object:
    .. code-block:: python
        @BACKBONE_REGISTRY.register()
        class MyBackbone():
            ...
    Or:
    .. code-block:: python
        BACKBONE_REGISTRY.register(MyBackbone)
    N)ÚnameÚreturncCs|
|_i|_
d
S)zI
        Args:
            name (str): the name of this registry
        N)Ú_nameÚ_obj_map)Úselfr©r
ú./utils/registry.pyÚ__init__s
zRegistry.__init__)rÚobjrcCs*|
|jkst
d
 |
|j¡ƒ
‚
||j|
<dS)Nz=An object named '{}' was already registered in '{}' registry!)rÚAssertionErrorÚformatr)r    rr r
r
rÚ_do_register sÿ
ÿþzRegistry._do_register)r rcs6|
d
kr ttdœ‡f
dd„}|S|
j
}ˆ ||
¡
d
S)z§
        Register the given object under the the name `obj.__name__`.
        Can be used as either a decorator or not. See docstring of this class for usage.
        N)Ú func_or_classrc
s|j}
ˆ 
|
|¡
|S)
N)Ú__name__r)rr©
r    r
rÚdeco/s


zRegistry.register.<locals>.deco)Úobjectrr)r    r rrr
rrÚregister(s

zRegistry.registercCs*|j 
|
¡
}|dkr&td
 |
|j¡ƒ
‚
|S)Nz,No object named '{}' found in '{}' registry!)rÚgetÚKeyErrorrr)r    rÚretr
r
rr:s




ÿÿzRegistry.get)
N)rÚ
__module__Ú__qualname__Ú__doc__Ústrrrrrrrr
r
r
rrs

rN)Útypingrrrrr
r
r
rÚ<module>s