39 lines
3.0 KiB
C++
39 lines
3.0 KiB
C++
#pragma once
|
|
#include "door_risk.hpp"
|
|
#include "pipeline_types.hpp"
|
|
namespace cms {
|
|
inline Json xy(std::vector<fangjia::PersonPose> const& people, int digits=3) {
|
|
Json a=Json::array(); for(auto const& p:people) { Json k=Json::array();for(auto const& v:p.keypoints) k.push_back({rounded(v.x,digits),rounded(v.y,digits)});a.push_back(k); } return a;
|
|
}
|
|
inline Json scores(std::vector<fangjia::PersonPose> const& people) {
|
|
Json a=Json::array();for(auto const& p:people) {Json k=Json::array();for(auto const& v:p.keypoints) k.push_back(v.score);a.push_back(k);}return a;
|
|
}
|
|
inline Json makeRecord(int frame,std::vector<fangjia::PersonPose> const& raw,std::vector<fangjia::PersonPose> const& isolated,fangjia::PoseFrameResult const& fused, bool diagnostics=true, bool visual=true) {
|
|
Json r={{"frame_index",frame},{"track_ids",Json::array()},{"boxes",Json::array()},{"det_confidences",Json::array()}};
|
|
for(auto const&p:raw) {auto const&b=p.box;r["track_ids"].push_back(b.trackId);r["boxes"].push_back({rounded(b.x1,3),rounded(b.y1,3),rounded(b.x2,3),rounded(b.y2,3)});r["det_confidences"].push_back(rounded(b.score,4));}
|
|
if(diagnostics) {
|
|
r["raw_keypoints"]=xy(raw,9);r["raw_scores"]=scores(raw);r["isolated_keypoints"]=xy(isolated,9);r["isolated_scores"]=scores(isolated);
|
|
r["inference_boxes"]=Json::array();for(auto const&p:raw)r["inference_boxes"].push_back({p.box.x1,p.box.y1,p.box.x2,p.box.y2});
|
|
}
|
|
if(visual||diagnostics) {
|
|
r["temporal_selected_keypoints"]=xy(fused.selectedPeople);r["temporal_keypoints"]=xy(fused.smoothedPeople);r["v2_keypoints"]=xy(fused.v2People);
|
|
r["temporal_metadata"]=Json::array();r["v2_metadata"]=Json::array();
|
|
for(size_t p=0;p<fused.metadata.size();++p) {
|
|
Json m=Json::object(),v=Json::object();
|
|
for(int s=0;s<2;++s) {auto side=s==0?"left":"right";auto const&t=fused.metadata[p][s];
|
|
m[side]={{"limb_state",t.limbConsensus?"consensus":"inferred"},{"hand_source",t.handSource},{"hand_state",t.handState},{"distance_eligible",t.distanceEligible},{"alarm_eligible",t.alarmEligible},{"selected_hand_valid",t.handValid},{"selected_hand_score",t.handScore}};
|
|
v[side]={{"state",fangjia::limbStateName(fused.limbs[p*2+s].state)}};
|
|
} r["temporal_metadata"].push_back(m);r["v2_metadata"].push_back(v);
|
|
}
|
|
}
|
|
r["global_hands"]=Json::array();
|
|
for(auto const& h:fused.globalHands) {
|
|
Json points=Json::array(),aliases=Json::array();double cx=0,cy=0;
|
|
for(auto const& k:h.keypoints) {points.push_back({rounded(k.x,3),rounded(k.y,3)});cx+=k.x;cy+=k.y;}
|
|
for(auto const&a:h.aliases) aliases.push_back({{"person_id",a.first},{"side",a.second?"left":"right"}});
|
|
r["global_hands"].push_back({{"person_id",h.personId},{"side",h.left?"left":"right"},{"source",h.source},{"state",h.state},{"distance_eligible",h.distanceEligible},{"alarm_eligible",h.alarmEligible},{"score",rounded(h.score,4)},{"center",{rounded(cx/21,3),rounded(cy/21,3)}},{"span_px",rounded(h.spanPx,3)},{"keypoints",points},{"aliases",aliases}});
|
|
}
|
|
return r;
|
|
}
|
|
}
|