Skip to content

Commit 3dc20b8

Browse files
committed
Wrong output format
1 parent 9f357c7 commit 3dc20b8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/torchcodec/_core/CudaDeviceInterface.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,8 @@ UniqueAVFrame CudaDeviceInterface::maybeConvertAVFrameToNV12(
244244
// n5.0. With the earlier version of ffmpeg we have no choice but use CPU
245245
// filters. See:
246246
// https://github.com/FFmpeg/FFmpeg/commit/62dc5df941f5e196164c151691e4274195523e95
247-
outputFormat = AV_PIX_FMT_RGB24;
247+
//outputFormat = AV_PIX_FMT_RGB24;
248+
outputFormat = AV_PIX_FMT_CUDA;
248249

249250
/*
250251
auto actualFormatName = av_get_pix_fmt_name(actualFormat);

0 commit comments

Comments
 (0)