413 lines
11 KiB
Python
413 lines
11 KiB
Python
annotationParametersMap = {
|
|
"Canny": {
|
|
"preprocessor": 1,
|
|
"annotationParameters": {
|
|
"canny": {
|
|
"preprocessorResolution": 512,
|
|
"lowThreshold": 100,
|
|
"highThreshold": 200
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"SoftEdge - hed": {
|
|
"preprocessor": 5,
|
|
"annotationParameters": {
|
|
"hed": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"SoftEdge - hed_safe": {
|
|
"preprocessor": 6,
|
|
"annotationParameters": {
|
|
"hedSafe": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"SoftEdge - pidinet": {
|
|
"preprocessor": 17,
|
|
"annotationParameters": {
|
|
"pidinet": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"SoftEdge - pidinet_safe": {
|
|
"preprocessor": 18,
|
|
"annotationParameters": {
|
|
"pidinetSafe": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"SoftEdge - softedge_teed": {
|
|
"preprocessor": 58,
|
|
"annotationParameters": {
|
|
"softedgeTeed": {
|
|
"preprocessorResolution": 512,
|
|
"safeSteps": 2
|
|
}
|
|
},
|
|
"model": "3e6860a3b9444f25ae07d9c1b5d1ba9e"
|
|
},
|
|
"SoftEdge - softedge_anyline": {
|
|
"preprocessor": 65,
|
|
"annotationParameters": {
|
|
"softedgeAnyline": {
|
|
"preprocessorResolution": 512,
|
|
"safeSteps": 2
|
|
}
|
|
},
|
|
"model": "3e6860a3b9444f25ae07d9c1b5d1ba9e"
|
|
},
|
|
# "MLSD（直线） - mlsd (M-LSD 直线线条检测)": {
|
|
# "preprocessor": 8,
|
|
# "annotationParameters": {
|
|
# "mlsd": {
|
|
# "preprocessorResolution": 512,
|
|
# "valueThreshold": 0.1,
|
|
# "distanceThreshold": 0.1
|
|
# }
|
|
# }
|
|
# },
|
|
"Scribble/Sketch - scribble_pidinet": {
|
|
"preprocessor": 20,
|
|
"annotationParameters": {
|
|
"scribblePidinet": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Scribble/Sketch - scribble_xdog": {
|
|
"preprocessor": 21,
|
|
"annotationParameters": {
|
|
"scribbleXdog": {
|
|
"preprocessorResolution": 512,
|
|
"XDoGThreshold": 32
|
|
}
|
|
}
|
|
},
|
|
"Scribble/Sketch - scribble_hed": {
|
|
"preprocessor": 22,
|
|
"annotationParameters": {
|
|
"scribbleHed": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Lineart- lineart_realistic": {
|
|
"preprocessor": 29,
|
|
"annotationParameters": {
|
|
"lineartRealistic": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Lineart - lineart standard": {
|
|
"preprocessor": 32,
|
|
"annotationParameters": {
|
|
"lineartStandard": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Lineart - lineart coarse": {
|
|
"preprocessor": 30,
|
|
"annotationParameters": {
|
|
"lineartCoarse": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Lineart - lineart_anime": {
|
|
"preprocessor": 31,
|
|
"annotationParameters": {
|
|
"lineartAnime": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Lineart - lineart_anime_denoise": {
|
|
"preprocessor": 36,
|
|
"annotationParameters": {
|
|
"lineartAnimeDenoise": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Depth - depth_midas": {
|
|
"preprocessor": 2,
|
|
"annotationParameters": {
|
|
"depthMidas": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Depth - depth_leres (LeRes)": {
|
|
"preprocessor": 3,
|
|
"annotationParameters": {
|
|
"depthLeres": {
|
|
"preprocessorResolution": 512,
|
|
"removeNear": 0,
|
|
"removeBackground": 0
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Depth - depth_leres++": {
|
|
"preprocessor": 4,
|
|
"annotationParameters": {
|
|
"depthLeresPlus": {
|
|
"preprocessorResolution": 512,
|
|
"removeNear": 0,
|
|
"removeBackground": 0
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Depth - depth_zoe (ZoE)": {
|
|
"preprocessor": 25,
|
|
"annotationParameters": {
|
|
"depthZoe": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Depth - depth_hand_refiner": {
|
|
"preprocessor": 57,
|
|
"annotationParameters": {
|
|
"depthHandRefiner": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"Depth - depth_anything": {
|
|
"preprocessor": 64,
|
|
"annotationParameters": {
|
|
"depthAnything": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
# "segmentation": {
|
|
# "preprocessor": 23,
|
|
# "annotationParameters": {
|
|
# "segmentation": {
|
|
# "preprocessorResolution": 512
|
|
# }
|
|
# }
|
|
# },
|
|
# "oneformer_coco": {
|
|
# "preprocessor": 27,
|
|
# "annotationParameters": {
|
|
# "oneformerCoco": {
|
|
# "preprocessorResolution": 512
|
|
# }
|
|
# }
|
|
# },
|
|
# "oneformer_ade20k": {
|
|
# "preprocessor": 28,
|
|
# "annotationParameters": {
|
|
# "oneformerAde20k": {
|
|
# "preprocessorResolution": 512
|
|
# }
|
|
# }
|
|
# },
|
|
# "anime_face_segment": {
|
|
# "preprocessor": 54,
|
|
# "annotationParameters": {
|
|
# "animeFaceSegment": {
|
|
# "preprocessorResolution": 512
|
|
# }
|
|
# }
|
|
# },
|
|
"Normal - normal_map": {
|
|
"preprocessor": 9,
|
|
"annotationParameters": {
|
|
"normalMap": {
|
|
"preprocessorResolution": 512,
|
|
"backgroundThreshold": 0.4
|
|
}
|
|
},
|
|
"model": "e51fdccdf3b8417aab246bde40b5f360"
|
|
},
|
|
"Normal - normal bae (Bae)": {
|
|
"preprocessor": 26,
|
|
"annotationParameters": {
|
|
"normalBae": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "e51fdccdf3b8417aab246bde40b5f360"
|
|
},
|
|
"OpenPose - mediapipe_face": {
|
|
"preprocessor": 7,
|
|
"annotationParameters": {
|
|
"mediapipeFace": {
|
|
"preprocessorResolution": 512,
|
|
"maxFaces": 1,
|
|
"minConfidence": 0.5
|
|
}
|
|
},
|
|
"model": "7c6d889cb9c04b78858d8fece80f9f85"
|
|
},
|
|
"OpenPose - openpose": {
|
|
"preprocessor": 10,
|
|
"annotationParameters": {
|
|
"openpose": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"OpenPose - openpose hand": {
|
|
"preprocessor": 11,
|
|
"annotationParameters": {
|
|
"openposeHand": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"OpenPose - openpose face": {
|
|
"preprocessor": 12,
|
|
"annotationParameters": {
|
|
"openposeFace": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"OpenPose - openpose_faceonly": {
|
|
"preprocessor": 13,
|
|
"annotationParameters": {
|
|
"openposeFaceonly": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"OpenPose - openpose_full": {
|
|
"preprocessor": 14,
|
|
"annotationParameters": {
|
|
"openposeFull": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
"OpenPose - dw_openpose_full": {
|
|
"preprocessor": 45,
|
|
"annotationParameters": {
|
|
"dwOpenposeFull": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "13c1e1b96ba64f9cbb2b54f89b5fe873"
|
|
},
|
|
# "OpenPose - animal_openpose": {
|
|
# "preprocessor": 53,
|
|
# "annotationParameters": {
|
|
# "animalOpenpose": {
|
|
# "preprocessorResolution": 512
|
|
# }
|
|
# }
|
|
# },
|
|
# "OpenPose - densepose_parula": {
|
|
# "preprocessor": 56,
|
|
# "annotationParameters": {
|
|
# "denseposeParula": {
|
|
# "preprocessorResolution": 512
|
|
# }
|
|
# }
|
|
# }
|
|
"Tile/Blur - tile_resample": {
|
|
"preprocessor": 34,
|
|
"annotationParameters": {
|
|
"tileResample": {
|
|
"downSamplingRate": 1
|
|
}
|
|
},
|
|
"model": "a696b5bdadc740119fd76505b33d6898"
|
|
},
|
|
"Tile/Blur - tile_colorfix": {
|
|
"preprocessor": 43,
|
|
"annotationParameters": {
|
|
"tileColorfix": {
|
|
"variation": 8
|
|
}
|
|
},
|
|
"model": "a696b5bdadc740119fd76505b33d6898"
|
|
},
|
|
"Tile/Blur - tile_colorfix+sharp": {
|
|
"preprocessor": 44,
|
|
"annotationParameters": {
|
|
"tileColorfixSharp": {
|
|
"variation": 8,
|
|
"sharpness": 1
|
|
}
|
|
},
|
|
"model": "a696b5bdadc740119fd76505b33d6898"
|
|
},
|
|
"Tile/Blur - blur_gaussian": {
|
|
"preprocessor": 52,
|
|
"annotationParameters": {
|
|
"blurGaussian": {
|
|
"preprocessorResolution": 512,
|
|
"sigma": 9
|
|
}
|
|
},
|
|
"model": "a696b5bdadc740119fd76505b33d6898"
|
|
},
|
|
"IP-Adapter - ip-adapter-siglip": {
|
|
"preprocessor": 66,
|
|
"annotationParameters": {
|
|
"ipAdapterSiglip": {
|
|
"preprocessorResolution": 512
|
|
}
|
|
},
|
|
"model": "c6ed70879cf011ef96d600163e37ec70"
|
|
},
|
|
"Inpaint - inpaint_global_harmonious": {
|
|
"preprocessor": 40,
|
|
"annotationParameters": {
|
|
"inpaintGlobalHarmonious": {}
|
|
},
|
|
"model": "31df01fc271d484ca4d496179d69a665"
|
|
},
|
|
"Inpaint - inpaint_only": {
|
|
"preprocessor": 41,
|
|
"annotationParameters": {
|
|
"inpaintOnly": {}
|
|
},
|
|
"model": "31df01fc271d484ca4d496179d69a665"
|
|
},
|
|
"Inpaint - inpaint_only+lama": {
|
|
"preprocessor": 42,
|
|
"annotationParameters": {
|
|
"inpaintOnlyLama": {}
|
|
},
|
|
"model": "31df01fc271d484ca4d496179d69a665"
|
|
}
|
|
}
|
|
|
|
def get_annotation_parameters_by_namer(preprocessor):
|
|
return annotationParametersMap[preprocessor]
|