Files
ali-vilab-ACE_plus/models/model_zoo.yaml
T
2025-01-05 20:37:47 +08:00

28 lines
884 B
YAML

MODEL:
PORTRAIT:
MODEL_PATH: ${PORTRAIT_MODEL_PATH}
SUBJECT:
MODEL_PATH: ${SUBJECT_MODEL_PATH}
LOCAL_EDITING:
MODEL_PATH: ${LOCAL_MODEL_PATH}
REPAINTING_SCALE: 0.5
PREPROCESSOR:
- NAME: CannyAnnotator
TYPE: canny_repainting
LOW_THRESHOLD: 100
HIGH_THRESHOLD: 200
- NAME: ColorAnnotator
TYPE: mosaic_repainting
RATIO: 64
- NAME: InfoDrawContourAnnotator
TYPE: contour_repainting
INPUT_NC: 3
OUTPUT_NC: 1
N_RESIDUAL_BLOCKS: 3
SIGMOID: True
PRETRAINED_MODEL: "ms://iic/scepter_annotator@annotator/ckpts/informative_drawing_contour_style.pth"
- NAME: MidasDetector
PRETRAINED_MODEL: "ms://iic/scepter_annotator@annotator/ckpts/dpt_hybrid-midas-501f0c75.pt"
TYPE: depth_repainting
- NAME: GrayAnnotator
TYPE: recolorizing