Mixed updates and improvements.

2025-01-21 16:23:19 +01:00
parent c5f190ab35
commit 29c072400f
11 changed files with 49 additions and 82 deletions
--- a/scripts/utils_2d_pose.py
+++ b/scripts/utils_2d_pose.py
@ -20,6 +20,7 @@ class BaseModel(ABC):
            raise FileNotFoundError("File not found:", model_path)

        if model_path.endswith(".onnx"):
+            print("Loading model:", model_path)
            self.init_onnxruntime(model_path)
            self.runtime = "ort"
        else: