main.py

import os
import cv2
print(f'OpenCV: {cv2.__version__} for python installed and working')
image = cv2.imread('bus.jpg')
if image is None:print("無法加載圖像1")
print(cv2.cuda.getCudaEnabledDeviceCount())
cv2.cuda.setDevice(0)
cv2.cuda.printCudaDeviceInfo(0)
image_gpu = cv2.cuda_GpuMat()
image_gpu.upload(image)
screenshot = cv2.cuda.cvtColor(image_gpu, cv2.COLOR_RGB2BGR)
screenshot = cv2.cuda.resize(image_gpu, (400, 400))
result_cpu = screenshot.download()
print("圖像寬度: ", image.shape)
print("數據類型:", image.dtype)
print("在CPU下，原始圖像格式:", type(image))
print("在GPU下，處理后的圖像格式:", type(screenshot))
print("在CPU下，處理后的圖像格式:", type(result_cpu))
print("圖像寬度: ", result_cpu.shape)
print("數據類型:", result_cpu.dtype)
if result_cpu is None:print("無法加載圖像2")
else:cv2.imshow("Window", result_cpu)cv2.waitKey(0)cv2.destroyAllWindows()

main.cpp

#include <iostream>
#include <opencv2/opencv.hpp>
#include <opencv2/cudaimgproc.hpp>
int main(int argc, char* argv[])
{cv::Mat h_img1 = cv::imread("bus.jpg");cv::cuda::GpuMat d_result1, d_result2, d_result3, d_result4, d_img1;d_img1.upload(h_img1);cv::cuda::cvtColor(d_img1, d_result1, cv::COLOR_BGR2GRAY);cv::cuda::cvtColor(d_img1, d_result2, cv::COLOR_BGR2RGB);cv::cuda::cvtColor(d_img1, d_result3, cv::COLOR_BGR2HSV);cv::cuda::cvtColor(d_img1, d_result4, cv::COLOR_BGR2YCrCb);cv::Mat h_result1, h_result2, h_result3, h_result4;d_result1.download(h_result1);d_result2.download(h_result2);d_result3.download(h_result3);d_result4.download(h_result4);cv::imshow("Gray處理結果：", h_result1);cv::imshow("RGB處理結果：", h_result2);cv::imshow("HSV處理結果：", h_result3);cv::imshow("YCrCb處理結果：", h_result4);cv::waitKey(0);return 0;
}

CMakeLists.txt

cmake_minimum_required(VERSION 3.10)
project(OpenCVCuda LANGUAGES CXX)
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set("OpenCV_DIR" "E:\\Opencv gpu\\newbuild\\install")
set(OpenCV_INCLUDE_DIRS ${OpenCV_DIR}\\include)
set(OpenCV_LIB_DEBUG ${OpenCV_DIR}\\x64\\vc17\\lib\\opencv_world470d.lib) 
set(OpenCV_LIB_RELEASE ${OpenCV_DIR}\\x64\\vc17\\lib\\opencv_world470.lib)   
set(CMAKE_CXX_STANDARD 14)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CUDA_ARCHITECTURES 86)		
find_package(CUDA REQUIRED)			
enable_language(CUDA)  
include_directories(${OpenCV_INCLUDE_DIRS}) 
link_directories(${OpenCV_LIB_DIRS})  
find_package(OpenCV QUIET)	
link_libraries(${OpenCV_LIBS})
add_executable(OpenCVCuda main.cpp)
target_link_libraries(${PROJECT_NAME} $<$<CONFIG:Debug>:${OpenCV_LIB_DEBUG}>$<$<CONFIG:Release>:${OpenCV_LIB_RELEASE}>
)
target_compile_features(OpenCVCuda PRIVATE cxx_std_14)