赞
踩
具体的api接口:
git clone -b cumtchw-patch-1 https://github.com/cumtchw/cuda_utils
下面是调用上述接口对图片进行resize的一个demo.
- #include <fstream>
- #include <iostream>
- #include <thread>
- #include <string>
- #include <vector>
- #include <cuda_runtime_api.h>
- #include "opencv2/opencv.hpp"
- #include <unistd.h>
-
- #include "cuda_utils_sdk.h"
- #include "cudaConverter.h"
-
- using namespace cv;
- using namespace std;
-
- int main(int argc, char ** argv)
- {
-
- int img_w_{};
- int img_h_{};
- int img_step1_{};
- vector<float> preprocess_output_data_;
- size_t convert_input_size_{};
- size_t convert_output_size_{};
- uchar* convert_input_{};
- float* convert_output_{};
- int m_maxBatchSize = 1;
- std::vector<float> mean_data_{ 0, 0, 0 };
- int w = 1024;
- int h = 1024;
- int c =3;
- int error{};
-
- cv::Mat src0 = cv::imread("./test0.
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。