CV-CUDA
CV-CUDA copied to clipboard
[BUG] The result of cvtColor RGB2YUV is not same with opencv
opencv code:
from pathlib import Path import numpy as np import cv2 import cvcuda import logging import torch from PIL import Image
img = cv2.imread('./0.jpg', cv2.IMREAD_UNCHANGED) pil_img = Image.fromarray(img.astype(np.uint8)) image1 = np.array(pil_img) image2 = cv2.cvtColor(image1, cv2.COLOR_RGB2BGR).astype(np.float32) cv_YUV = cv2.cvtColor(image2, cv2.COLOR_BGR2YUV)
cvcuda
image = torch.from_numpy(image1).to(torch.uint8).cuda() frame_hwc = cvcuda.as_tensor(image .cuda(), "HWC") cvcuda_YUV = cvcuda.cvtcolor(frame_hwc, cvcuda.ColorConversion.RGB2YUV)
Describe the bug cv_YUV != cvcuda_YUV
Expected behavior cv_YUV == cvcuda_YUV