File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -598,8 +598,8 @@ void VideoEncoder::initializeEncoder(
598
598
// TODO-VideoEncoder: Remove assumption that tensor in NCHW format
599
599
auto sizes = frames_.sizes ();
600
600
inPixelFormat_ = AV_PIX_FMT_GBRP;
601
- inHeight_ = sizes[2 ];
602
- inWidth_ = sizes[3 ];
601
+ inHeight_ = static_cast < int >( sizes[2 ]) ;
602
+ inWidth_ = static_cast < int >( sizes[3 ]) ;
603
603
604
604
// Use specified dimensions or input dimensions
605
605
// TODO-VideoEncoder: Allow height and width to be set
@@ -664,7 +664,7 @@ void VideoEncoder::encode() {
664
664
getFFMPEGErrorStringFromErrorCode (status));
665
665
666
666
AutoAVPacket autoAVPacket;
667
- int numFrames = frames_.sizes ()[0 ];
667
+ int numFrames = static_cast < int >( frames_.sizes ()[0 ]) ;
668
668
for (int i = 0 ; i < numFrames; ++i) {
669
669
torch::Tensor currFrame = frames_[i];
670
670
UniqueAVFrame avFrame = convertTensorToAVFrame (currFrame, i);
You can’t perform that action at this time.
0 commit comments