Yolo-Detection/upbot_SORT/include/sort_test/preprocess.h

17 lines
610 B
C
Raw Permalink Normal View History

2024-11-21 14:16:10 +08:00
#ifndef _upbot_vision_PREPROCESS_H_
#define _upbot_vision_PREPROCESS_H_
#include <stdio.h>
#include "im2d.h"
#include "rga.h"
#include "opencv2/core/core.hpp"
#include "opencv2/imgcodecs.hpp"
#include "opencv2/imgproc.hpp"
#include "sort_test/postprocess.h"
void letterbox(const cv::Mat &image, cv::Mat &padded_image, BOX_RECT &pads, const float scale, const cv::Size &target_size, const cv::Scalar &pad_color = cv::Scalar(128, 128, 128));
int resize_rga(rga_buffer_t &src, rga_buffer_t &dst, const cv::Mat &image, cv::Mat &resized_image, const cv::Size &target_size);
#endif //_upbot_vision_PREPROCESS_H_