Fxi path

Files changed (2) hide show

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

@@ -513,8 +513,10 @@ class Wholebody:
         providers = (
             ["CPUExecutionProvider"] if device == "cpu" else ["CUDAExecutionProvider"]
         )
-        onnx_det = ModelDataPathPrefix.joinpath("DWPose/yolox_l.onnx")
-        onnx_pose = ModelDataPathPrefix.joinpath("DWPose/dw-ll_ucoco_384.onnx")
         self.session_det = ort.InferenceSession(
             path_or_bytes=onnx_det, providers=providers

         providers = (
             ["CPUExecutionProvider"] if device == "cpu" else ["CUDAExecutionProvider"]
         )
+        base_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+        onnx_det = os.path.join(base_dir, "pretrained_weights", "DWPose", "yolox_l.onnx")
+        onnx_pose = os.path.join(base_dir, "pretrained_weights", "DWPose", "dw-ll_ucoco_384.onnx")
         self.session_det = ort.InferenceSession(
             path_or_bytes=onnx_det, providers=providers