add specific sfd path

tongtongking · Jun 11, 2024 · 98bf0e4 · 98bf0e4
1 parent 028fb80
commit 98bf0e4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Musetalk/musetalk/utils/face_detection/detection/sfd/sfd_detector.py b/Musetalk/musetalk/utils/face_detection/detection/sfd/sfd_detector.py
@@ -14,7 +14,7 @@
 
 
 class SFDDetector(FaceDetector):
-    def __init__(self, device, path_to_detector=os.path.join(os.path.dirname(os.path.abspath(__file__)), 's3fd.pth'), verbose=False):
+    def __init__(self, device, path_to_detector='checkpoints/hub/checkpoints/s3fd-619a316812.pth', verbose=False):
         super(SFDDetector, self).__init__(device, verbose)
 
         # Initialise the face detector

diff --git a/face_detection/detection/sfd/sfd_detector.py b/face_detection/detection/sfd/sfd_detector.py
@@ -14,7 +14,7 @@
 
 
 class SFDDetector(FaceDetector):
-    def __init__(self, device, path_to_detector=os.path.join(os.path.dirname(os.path.abspath(__file__)), 's3fd.pth'), verbose=False):
+    def __init__(self, device, path_to_detector=r'checkpoints/hub/checkpoints/s3fd-619a316812.pth', verbose=False):
         super(SFDDetector, self).__init__(device, verbose)
 
         # Initialise the face detector