Improve GPU frames accuracy (#413)

pytorch · Dec 2, 2024 · de6a6d4 · de6a6d4
1 parent ce99662
commit de6a6d4
Show file tree

Hide file tree

Showing 5 changed files with 92 additions and 137 deletions.
diff --git a/src/torchcodec/decoders/_core/CudaDevice.cpp b/src/torchcodec/decoders/_core/CudaDevice.cpp
@@ -225,7 +225,7 @@ void convertAVFrameToDecodedOutputOnCuda(
   auto start = std::chrono::high_resolution_clock::now();
   NppStatus status;
   if (src->colorspace == AVColorSpace::AVCOL_SPC_BT709) {
-    status = nppiNV12ToRGB_709HDTV_8u_P2C3R(
+    status = nppiNV12ToRGB_709CSC_8u_P2C3R(
         input,
         src->linesize[0],
         static_cast<Npp8u*>(dst.data_ptr()),