Skip to content

Commit 4eec101

Browse files
author
zyan3
committed
Merge branch 'torchvision_video_reader' of github.com:stephenyan1231/vision into torchvision_video_reader
2 parents b3f2a6e + 38c0e75 commit 4eec101

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

torchvision/csrc/cpu/video_reader/VideoReader.h

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
// Interface for Python
66

7+
<<<<<<< HEAD
78
/*
89
return:
910
videoFrame: tensor (N, H, W, C) kByte
@@ -15,6 +16,8 @@
1516
audioTimeBase: tensor (2) kInt
1617
audioSampleRate: tensor (1) kInt
1718
*/
19+
=======
20+
>>>>>>> 38c0e753cf22326e10bfb992bce70bff95059253
1821
torch::List<torch::Tensor> readVideoFromMemory(
1922
// 1D tensor of data type uint8, storing the comparessed video data
2023
torch::Tensor input_video,
@@ -71,7 +74,10 @@ torch::List<torch::Tensor> readVideoFromMemory(
7174
int64_t audioTimeBaseNum,
7275
int64_t audioTimeBaseDen);
7376

77+
<<<<<<< HEAD
7478

79+
=======
80+
>>>>>>> 38c0e753cf22326e10bfb992bce70bff95059253
7581
torch::List<torch::Tensor> readVideoFromFile(
7682
std::string videoPath,
7783
double seekFrameMargin,

0 commit comments

Comments
 (0)