diff --git a/src/PicAnalysis.cpp b/src/PicAnalysis.cpp index 52a3e81..d1fc2ba 100644 --- a/src/PicAnalysis.cpp +++ b/src/PicAnalysis.cpp @@ -330,7 +330,7 @@ vector PicAnalysis::analysis_img(vector vec_img){ std::vector vec_hcp_result = m_human_car_algorithm.detect(vec_motor_img); if (vec_hcp_result.size() > 0 && road_info.vec_direct.size() > 0) { - int head_or_tail = vec_hcp_result[11].type; + int head_or_tail = vec_hcp_result[0].orient; if (head_or_tail == 0 || head_or_tail == 1) { result_info.reverse_driving = m_road_seg_algorithm.check_reverse_driving(road_info.vec_direct, vehicle_rect, src.width, src.height, head_or_tail); } diff --git a/src/ai_engine_module/HumanCarAnalysis.cpp b/src/ai_engine_module/HumanCarAnalysis.cpp index dd6ef45..9125318 100644 --- a/src/ai_engine_module/HumanCarAnalysis.cpp +++ b/src/ai_engine_module/HumanCarAnalysis.cpp @@ -52,12 +52,37 @@ std::vector HumanCarAnalysis::detect(vector vec_img){ } for(int batchIdx = 0;batchIdx