Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

cudacodec::Videoreader fix YUV color conversion #3828

Open
wants to merge 1 commit into
base: 4.x
Choose a base branch
from

Conversation

cudawarped
Copy link
Contributor

@cudawarped cudawarped commented Nov 22, 2024

Currently VideoReader color conversion:

  1. Forces quantization to 8 bits by using the NV12 surface format for all inputs regardless of the encoded bit depth or chroma subsampling.
  2. Does not respect the color space standard if specified in the video source.
  3. Only outputs BGR, BGRA, GRAY and NV12 frames.

Note: Previous attempts to fix this #3468 didn't address (2).

This PR uses the Nvidia Video Codec SDK ColorSpace.cu sample as a guide to:

  1. Use the color space standard if it exists and default back to a suitable format based on the video resolution if not.
  2. Decode YUV420 10/12 bit and YUV444 8/10/12 bit sources.
  3. Allow the user to choose the desired output bitdepth. e.g. Reqest 8 bit output from a 16 bit source.
  4. Add RGB/RGBA and planar outputs.

Pull Request Readiness Checklist

See details at https://github.com/opencv/opencv/wiki/How_to_contribute#making-a-good-pull-request

  • I agree to contribute to the project under Apache 2 License.
  • To the best of my knowledge, the proposed patch is not based on a code under GPL or another license that is incompatible with OpenCV
  • The PR is proposed to the proper branch
  • There is a reference to the original bug report and related work
  • There is accuracy test, performance test and test data in opencv_extra repository, if applicable
    Patch to opencv_extra has the same branch name.
  • The feature is well documented and sample code can be built with the project CMake

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants