| 73 | } |
| 74 | |
| 75 | void YOLO_NAS::preprocess(const std::vector<cv::Mat>& imgsBatch) |
| 76 | { |
| 77 | resizeDevice(m_param.batch_size, m_input_src_device, m_param.src_w, m_param.src_h, |
| 78 | m_input_resize_device, m_resize_shape.width, m_resize_shape.height, 114, m_dst2src); |
| 79 | copyWithPaddingDevice(m_param.batch_size, m_input_resize_device, m_resize_shape.width, m_resize_shape.height, |
| 80 | m_input_resize_padding_device, m_param.dst_w, m_param.dst_h, 114.f, m_pad_top, m_pad_left); |
| 81 | bgr2rgbDevice(m_param.batch_size, m_input_resize_padding_device, m_param.dst_w, m_param.dst_h, |
| 82 | m_input_rgb_device, m_param.dst_w, m_param.dst_h); |
| 83 | normDevice(m_param.batch_size, m_input_rgb_device, m_param.dst_w, m_param.dst_h, |
| 84 | m_input_norm_device, m_param.dst_w, m_param.dst_h, m_param); |
| 85 | hwc2chwDevice(m_param.batch_size, m_input_norm_device, m_param.dst_w, m_param.dst_h, |
| 86 | m_input_hwc_device, m_param.dst_w, m_param.dst_h); |
| 87 | } |
| 88 | void YOLO_NAS::postprocess(const std::vector<cv::Mat>& imgsBatch) |
| 89 | { |
| 90 | yolo_nas::decodeDevice(m_param, m_output_src_device, 4 + m_param.num_class, m_total_objects, m_output_area, |