| 201 | return out |
| 202 | |
| 203 | class InstantIDFaceAnalysis: |
| 204 | @classmethod |
| 205 | def INPUT_TYPES(s): |
| 206 | return { |
| 207 | "required": { |
| 208 | "provider": (["CPU", "CUDA", "ROCM", "CoreML"], ), |
| 209 | }, |
| 210 | } |
| 211 | |
| 212 | RETURN_TYPES = ("FACEANALYSIS",) |
| 213 | FUNCTION = "load_insight_face" |
| 214 | CATEGORY = "InstantID" |
| 215 | |
| 216 | def load_insight_face(self, provider): |
| 217 | model = FaceAnalysis(name="antelopev2", root=INSIGHTFACE_DIR, providers=[provider + 'ExecutionProvider',]) # alternative to buffalo_l |
| 218 | model.prepare(ctx_id=0, det_size=(640, 640)) |
| 219 | |
| 220 | return (model,) |
| 221 | |
| 222 | class FaceKeypointsPreprocessor: |
| 223 | @classmethod |
nothing calls this directly
no outgoing calls
no test coverage detected