File size: 1,359 Bytes
11ccb1b
 
 
 
 
 
 
 
75889ad
 
 
 
 
 
 
 
 
 
6a80886
11ccb1b
 
75889ad
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6a80886
 
11ccb1b
 
 
 
 
 
 
 
 
 
 
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
41
42
43
44
45
46
47
48
49
50
51
from .canny import CannyDetector
from .openpose import OpenposeDetector
from .binary import BinaryDetector
from .hed import HedDetector
# from .keypose import KeyPoseProcess
from .midas import MidasProcessor
from .mlsd import MLSDProcessor
from .uniformer import UniformerDetector
from .lineart import LineArtDetector
from .lineart_anime import LineArtAnimeDetector
from .manga_line import MangaLineExtration
from .leres import LeresPix2Pix
from .mediapipe_face import MediaPipeFace
from .normalbae import NormalBaeDetector
from .pidinet import PidInet
from .shuffle import Image2MaskShuffleDetector
from .zoe import ZoeDetector
from .oneformer import OneformerDetector
from .denthPose import DenthPoseProcessor

__all__ = [
    UniformerDetector,
    HedDetector,
    MLSDProcessor,
    BinaryDetector,
    CannyDetector,
    OpenposeDetector,
    MidasProcessor,
    LineArtDetector,
    LineArtAnimeDetector,
    MangaLineExtration,
    LeresPix2Pix,
    MediaPipeFace,
    NormalBaeDetector,
    PidInet,
    Image2MaskShuffleDetector,
    ZoeDetector,
    OneformerDetector,
    DenthPoseProcessor
]
#
#
# # default cache
# default_home = os.path.join(os.path.expanduser("~"), ".cache")
# model_cache_home = os.path.expanduser(
#     os.getenv(
#         "HF_HOME",
#         os.path.join(os.getenv("XDG_CACHE_HOME", default_home), "model"),
#     )
# )