From 291afb9c81aae1d15d9e3edcea618d0a2e1fc5a6 Mon Sep 17 00:00:00 2001
From: wangrong <wangrong@shsening.com>
Date: 星期三, 19 三月 2025 17:51:38 +0800
Subject: [PATCH] add samples
---
nodes/ba/vp_ba_stop_node.cpp | 31 +++++++++++++++++--------------
1 files changed, 17 insertions(+), 14 deletions(-)
diff --git a/nodes/ba/vp_ba_stop_node.cpp b/nodes/ba/vp_ba_stop_node.cpp
index e633d29..9ae20af 100644
--- a/nodes/ba/vp_ba_stop_node.cpp
+++ b/nodes/ba/vp_ba_stop_node.cpp
@@ -59,22 +59,25 @@
// for vp_frame_target only
std::vector<int> hit_traget_ids;
for (auto& target : meta->targets) {
- auto len = target->tracks.size();
- auto loc = target->get_rect().track_point();
+ if (target->primary_label == target->class_label_of_car)
+ {
+ auto len = target->tracks.size();
+ auto loc = target->get_rect().track_point();
- // target has been tracked AND tracked enough frames
- if (len < check_interval_frames || target->track_id < 0) {
- continue;
- }
- // if target inside of stop region or not
- if (!point_in_poly(loc, stop_region)) {
- continue;
- }
+ // target has been tracked AND tracked enough frames
+ if (len < check_interval_frames || target->track_id < 0) {
+ continue;
+ }
+ // if target inside of stop region or not
+ if (!point_in_poly(loc, stop_region)) {
+ continue;
+ }
- auto pre_loc = target->tracks[len - check_interval_frames].track_point();
- if (pre_loc.distance_with(loc) <= check_max_distance) {
- stop_checking_status[target->track_id]++;
- hit_traget_ids.push_back(target->track_id);
+ auto pre_loc = target->tracks[len - check_interval_frames].track_point();
+ if (pre_loc.distance_with(loc) <= check_max_distance) {
+ stop_checking_status[target->track_id]++;
+ hit_traget_ids.push_back(target->track_id);
+ }
}
}
--
Gitblit v1.9.1