@@ -564,7 +564,7 @@ torch::List<torch::Tensor> probeVideo(
564
564
565
565
} // namespace
566
566
567
- torch::List<torch::Tensor> readVideoFromMemory (
567
+ torch::List<torch::Tensor> read_video_from_memory (
568
568
torch::Tensor input_video,
569
569
double seekFrameMargin,
570
570
int64_t getPtsOnly,
@@ -608,7 +608,7 @@ torch::List<torch::Tensor> readVideoFromMemory(
608
608
audioTimeBaseDen);
609
609
}
610
610
611
- torch::List<torch::Tensor> readVideoFromFile (
611
+ torch::List<torch::Tensor> read_video_from_file (
612
612
std::string videoPath,
613
613
double seekFrameMargin,
614
614
int64_t getPtsOnly,
@@ -653,20 +653,20 @@ torch::List<torch::Tensor> readVideoFromFile(
653
653
audioTimeBaseDen);
654
654
}
655
655
656
- torch::List<torch::Tensor> probeVideoFromMemory (torch::Tensor input_video) {
656
+ torch::List<torch::Tensor> probe_video_from_memory (torch::Tensor input_video) {
657
657
return probeVideo (false , input_video, " " );
658
658
}
659
659
660
- torch::List<torch::Tensor> probeVideoFromFile (std::string videoPath) {
660
+ torch::List<torch::Tensor> probe_video_from_file (std::string videoPath) {
661
661
torch::Tensor dummy_input_video = torch::ones ({0 });
662
662
return probeVideo (true , dummy_input_video, videoPath);
663
663
}
664
664
665
665
TORCH_LIBRARY_FRAGMENT (video_reader, m) {
666
- m.def (" read_video_from_memory" , readVideoFromMemory );
667
- m.def (" read_video_from_file" , readVideoFromFile );
668
- m.def (" probe_video_from_memory" , probeVideoFromMemory );
669
- m.def (" probe_video_from_file" , probeVideoFromFile );
666
+ m.def (" read_video_from_memory" , read_video_from_memory );
667
+ m.def (" read_video_from_file" , read_video_from_file );
668
+ m.def (" probe_video_from_memory" , probe_video_from_memory );
669
+ m.def (" probe_video_from_file" , probe_video_from_file );
670
670
}
671
671
672
672
} // namespace video_reader
0 commit comments