| | |
| | | |
| | | #include <basic/debug/Debug.h> |
| | | #include "VptDetectWrapper.h" |
| | | #include <basic/timer_counter/Clocktimer.h> |
| | | |
| | | VptDetectWrapper::VptDetectWrapper() { |
| | | init(); |
| | |
| | | } |
| | | |
| | | list<VptDetectResults> VptDetectWrapper::process_image(cv::Mat &_img) { |
| | | ClockTimer ct("VptDetectWrapper::process_image"); |
| | | m_results.clear(); |
| | | |
| | | sy_img img; |
| | |
| | | int colorIndex = result.info[i].obj_index; |
| | | |
| | | if (colorIndex < 0 || colorIndex >= 9)continue; |
| | | if(colorIndex==4) DBG("detect a car, score =" << result.info[i].obj_score); |
| | | // sprintf(str_i, "%s:%.2f", type[colorIndex].c_str(), result.info[i].obj_score); |
| | | |
| | | // rectangle(process_image, |
| | |
| | | delete[] result.info; |
| | | result.info = NULL; |
| | | } |
| | | DBG(m_results.size()); |
| | | // DBG(m_results.size()); |
| | | return m_results; |
| | | } |
| | | |
| | |
| | | auto &carRes = vptDetectResults.vptDetectRes.carDetectRes; |
| | | |
| | | // 车颜色 |
| | | if(m_param.vehicle_color_config == SY_CONFIG_OPEN) { |
| | | carRes.colorLabel = static_cast<ColorLabel>(cur_res->vc_res.res_index); |
| | | carRes.colorLabelProb = cur_res->vc_res.res_prob; |
| | | } |
| | | |
| | | // 车型识别 |
| | | if(m_param.vehicle_recg_config == SY_CONFIG_OPEN) { |
| | | if (cur_res->vr_res.vehicle_brand != NULL) { |
| | | cout << " 车辆品牌 - " << cur_res->vr_res.vehicle_brand << endl; |
| | | carRes.vehicle_brand = cur_res->vr_res.vehicle_brand; |
| | |
| | | } |
| | | // cout << " " << cur_res->vr_res.name_score << endl; |
| | | carRes.vehicle_score = cur_res->vr_res.name_score; |
| | | } |
| | | |
| | | if(m_param.vehicle_pendant_det_recg_config == SY_CONFIG_OPEN) { |
| | | // 车牌 |
| | | if (cur_res->vp_res.rect.height_ != 0 && cur_res->vp_res.rect.width_ != 0) { |
| | | cout << "车牌位置: " << cur_res->vp_res.rect.left_ << " " << cur_res->vp_res.rect.top_ << " " |