| 184 | } |
| 185 | |
| 186 | void yolo::YOLO::preprocess(const std::vector<cv::Mat>& imgsBatch) |
| 187 | { |
| 188 | resizeDevice(m_param.batch_size, m_input_src_device, m_param.src_w, m_param.src_h, |
| 189 | m_input_resize_device, m_param.dst_w, m_param.dst_h, 114, m_dst2src); |
| 190 | bgr2rgbDevice(m_param.batch_size, m_input_resize_device, m_param.dst_w, m_param.dst_h, |
| 191 | m_input_rgb_device, m_param.dst_w, m_param.dst_h); |
| 192 | normDevice(m_param.batch_size, m_input_rgb_device, m_param.dst_w, m_param.dst_h, |
| 193 | m_input_norm_device, m_param.dst_w, m_param.dst_h, m_param); |
| 194 | hwc2chwDevice(m_param.batch_size, m_input_norm_device, m_param.dst_w, m_param.dst_h, |
| 195 | m_input_hwc_device, m_param.dst_w, m_param.dst_h); |
| 196 | } |
| 197 | |
| 198 | bool yolo::YOLO::infer() |
| 199 | { |