Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-05-01 11:41:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: mediapipe partial error

Browse Source
This commit is contained in:
Bingsu
2023-05-30 09:39:28 +09:00
parent 5ab196778b
commit ab3567aac6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
adetailer/mediapipe.py
View File

@@ -13,8 +13,8 @@ def mediapipe_predict(
model_type: str, image: Image.Image, confidence: float = 0.3 model_type: str, image: Image.Image, confidence: float = 0.3
) -> PredictOutput: ) -> PredictOutput:
mapping = { mapping = {
"mediapipe_face_short": partial(mediapipe_face_detection, model_type=0), "mediapipe_face_short": partial(mediapipe_face_detection, 0),
"mediapipe_face_full": partial(mediapipe_face_detection, model_type=1), "mediapipe_face_full": partial(mediapipe_face_detection, 1),
"mediapipe_face_mesh": mediapipe_face_mesh, "mediapipe_face_mesh": mediapipe_face_mesh,
} }
if model_type in mapping: if model_type in mapping: