增加一些耗时操作打印

2026-02-03 13:33:41 +08:00
parent 7f47735e0d
commit f04a8c0773
2 changed files with 33 additions and 5 deletions
--- a/algorithm/checkpoint/npu_yolo_pose_onnx.py
+++ b/algorithm/checkpoint/npu_yolo_pose_onnx.py
@@ -246,6 +246,7 @@ class YOLOv8_Pose_ONNX:
    # -------------------------------------------------
    def __call__(self, frame):
        inp = self.preprocess(frame)
+        # 耗时操作
        pred = self.session.run(None, {self.input_name: inp})[0]
        return self.postprocess(pred, frame.shape[:2])