Back to RGB24

Author: scotts

src/torchcodec/_core/CudaDeviceInterface.cpp

@@ -244,18 +244,15 @@ UniqueAVFrame CudaDeviceInterface::maybeConvertAVFrameToNV12(
     // n5.0. With the earlier version of ffmpeg we have no choice but use CPU
     // filters. See:
     // https://github.com/FFmpeg/FFmpeg/commit/62dc5df941f5e196164c151691e4274195523e95
-    //outputFormat = AV_PIX_FMT_RGB24;
-    outputFormat = AV_PIX_FMT_CUDA;
+    outputFormat = AV_PIX_FMT_RGB24;
 
-    /*
     auto actualFormatName = av_get_pix_fmt_name(actualFormat);
     TORCH_CHECK(
         actualFormatName != nullptr,
         "The actual format of a frame is unknown to FFmpeg. "
         "That's unexpected, please report this to the TorchCodec repo.");
 
-    */
-    filters << "hwupload,format=nv12";
+    filters << "hwdownload,format=" << actualFormatName;
   } else {
     // Actual output color format will be set via filter options
     outputFormat = AV_PIX_FMT_CUDA;