We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 15e23f6 commit 2342837Copy full SHA for 2342837
src/torchcodec/_core/Encoder.cpp
@@ -912,11 +912,6 @@ UniqueAVFrame VideoEncoder::convertTensorToAVFrame(
912
const torch::Tensor& frame,
913
int frameIndex) {
914
TORCH_CHECK(frame.is_cpu(), "CPU encoder requires CPU tensors");
915
- TORCH_CHECK(
916
- frame.dim() == 3 && frame.size(0) == 3,
917
- "Expected 3D RGB tensor (CHW format), got shape: ",
918
- frame.sizes());
919
-
920
// Initialize and cache scaling context if it does not exist
921
if (!swsContext_) {
922
swsContext_.reset(sws_getContext(
0 commit comments