Spaces:
Building
on
A10G
Building
on
A10G
File size: 1,114 Bytes
cab0202 |
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 |
from controlnet_aux import (
CannyDetector,
ContentShuffleDetector,
HEDdetector,
LineartAnimeDetector,
LineartDetector,
MediapipeFaceDetector,
MidasDetector,
MLSDdetector,
NormalBaeDetector,
OpenposeDetector,
PidiNetDetector,
SamDetector,
ZoeDetector,
)
PREPROCCES_DICT = {
"Hed": HEDdetector.from_pretrained("lllyasviel/Annotators"),
"Midas": MidasDetector.from_pretrained("lllyasviel/Annotators"),
"MLSD": MLSDdetector.from_pretrained("lllyasviel/Annotators"),
"Openpose": OpenposeDetector.from_pretrained("lllyasviel/Annotators"),
"PidiNet": PidiNetDetector.from_pretrained("lllyasviel/Annotators"),
"NormalBae": NormalBaeDetector.from_pretrained("lllyasviel/Annotators"),
"Lineart": LineartDetector.from_pretrained("lllyasviel/Annotators"),
"LineartAnime": LineartAnimeDetector.from_pretrained(
"lllyasviel/Annotators"
),
"Zoe": ZoeDetector.from_pretrained("lllyasviel/Annotators"),
"Canny": CannyDetector(),
"ContentShuffle": ContentShuffleDetector(),
"MediapipeFace": MediapipeFaceDetector(),
}
|