CV-CUDA icon indicating copy to clipboard operation
CV-CUDA copied to clipboard

[BUG] The result of cvtColor RGB2YUV is not same with opencv

Open hezeli123 opened this issue 9 months ago • 0 comments

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

hezeli123 avatar May 22 '24 12:05 hezeli123