We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 032ce75 commit 9fa7403Copy full SHA for 9fa7403
csrc/mmdeploy/apis/python/pose_tracker.cpp
@@ -30,7 +30,7 @@ std::vector<py::tuple> Apply(mmdeploy::PoseTracker* self,
30
std::vector<py::tuple> batch_ret;
31
batch_ret.reserve(frames.size());
32
for (const auto& rs : results) {
33
- py::array_t<float> keypoints({static_cast<int>(rs.size()), rs[0].keypoint_count, 3});
+ py::array_t<float> keypoints({static_cast<int>(rs.size()), rs.size() > 0 ? rs[0].keypoint_count : 0, 3});
34
py::array_t<float> bboxes({static_cast<int>(rs.size()), 4});
35
py::array_t<uint32_t> track_ids(static_cast<int>(rs.size()));
36
auto kpts_ptr = keypoints.mutable_data();
0 commit comments