修正抠图坐标全部都是0的问题

Hu Chunming
1 parent d15d6736
Showing 1 changed file with 7 additions and 2 deletions
src/ai_engine_module/pedestrian_vehicle_trespass.cpp
@@ -281,12 +281,17 @@ bool PedestrianVehicleTrespass::update_mstreams(const std::vector&lt;task_id_t&gt; &amp;ta
     // 221009 byzsh记录10条即可--------------------------------------------------------------------------------------
     if (obj_to_alarm_boxes_.find(obj_key) != obj_to_alarm_boxes_.end() && obj_to_alarm_boxes_[obj_key].size() >= 10)
       continue;
-    //--------------------------------------------------------------------------------------------------------------
+    
+    auto it_obj = obj_to_position_.find(obj_key);
+    if(it_obj == obj_to_position_.end()) {
+      continue;
+    }
+
     const unsigned stream_idx = it->second;
     auto &src_img = det_input_images[stream_idx];
     result_data_t result_data;
-    result_data.box = obj_to_position_[obj_key]; 
+    result_data.box = it_obj->second; 
     // 原图
     vpc_img_info src_img_info = VPCUtil::vpc_devMem2vpcImg(src_img);