fix

Files changed (4) hide show

__pycache__/handler.cpython-310.pyc CHANGED Viewed

Binary files a/__pycache__/handler.cpython-310.pyc and b/__pycache__/handler.cpython-310.pyc differ

extract_dwpose_from_vid.py CHANGED Viewed

@@ -6,7 +6,6 @@ from pathlib import Path
 import numpy as np
 import sys
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'src')))
 from src.dwpose import DWposeDetector
 from src.utils.util import get_fps, read_frames, save_videos_from_pil

 import numpy as np
 import sys
 from src.dwpose import DWposeDetector
 from src.utils.util import get_fps, read_frames, save_videos_from_pil

src/dwpose/__pycache__/wholebody.cpython-310.pyc CHANGED Viewed

Binary files a/src/dwpose/__pycache__/wholebody.cpython-310.pyc and b/src/dwpose/__pycache__/wholebody.cpython-310.pyc differ

src/dwpose/wholebody.py CHANGED Viewed

@@ -5,11 +5,16 @@ import cv2
 import numpy as np
 import onnxruntime as ort
-from .onnxdet import inference_detector
-from .onnxpose import inference_pose
-ModelDataPathPrefix = Path("./pretrained_weights")
 class Wholebody:
     def __init__(self, device="cuda:0"):

 import numpy as np
 import onnxruntime as ort
+import os
+import sys
+file_dir = os.path.dirname(__file__)
+sys.path.append(file_dir)
+from onnxdet import inference_detector
+from onnxpose import inference_pose
+ModelDataPathPrefix = Path("./pretrained_weights")
 class Wholebody:
     def __init__(self, device="cuda:0"):