diff --git a/csrc/mmdeploy/apis/python/pose_tracker.cpp b/csrc/mmdeploy/apis/python/pose_tracker.cpp index fea6895ae8..16c79bbfa4 100644 --- a/csrc/mmdeploy/apis/python/pose_tracker.cpp +++ b/csrc/mmdeploy/apis/python/pose_tracker.cpp @@ -30,7 +30,7 @@ std::vector Apply(mmdeploy::PoseTracker* self, std::vector batch_ret; batch_ret.reserve(frames.size()); for (const auto& rs : results) { - py::array_t keypoints({static_cast(rs.size()), rs[0].keypoint_count, 3}); + py::array_t keypoints({static_cast(rs.size()), rs.size() > 0 ? rs[0].keypoint_count : 0, 3}); py::array_t bboxes({static_cast(rs.size()), 4}); py::array_t track_ids(static_cast(rs.size())); auto kpts_ptr = keypoints.mutable_data();