__init__.py 961 Bytes
Newer Older
1
from .alexnet import AlexNet_simplified
2
3
4
from .arcface import ArcFaceLayer
from .arcface import ArcFaceLayer3Penalties
from .arcface import ArcFaceModel
5
from .densenet import DeepPixBiS
6
from .densenet import DenseNet
7
from .densenet import densenet161  # noqa: F401
8
from .embedding_validation import EmbeddingValidation
9
10
from .mine import MineModel

Amir MOHAMMADI's avatar
Amir MOHAMMADI committed
11

12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# gets sphinx autodoc done right - don't remove it
def __appropriate__(*args):
    """Says object was actually declared here, an not on the import module.

    Parameters:

      *args: An iterable of objects to modify

    Resolves `Sphinx referencing issues
    <https://github.com/sphinx-doc/sphinx/issues/3048>`
    """

    for obj in args:
        obj.__module__ = __name__


28
29
30
31
32
33
34
35
36
37
__appropriate__(
    AlexNet_simplified,
    DenseNet,
    DeepPixBiS,
    MineModel,
    ArcFaceLayer,
    ArcFaceLayer3Penalties,
    ArcFaceModel,
    EmbeddingValidation,
)
38
__all__ = [_ for _ in dir() if not _.startswith("_")]