【文件属性】:
文件名称:load_video_as_tensor:尝试使用不同的软件包将视频加载到PyTorch中的Tensor中
文件大小:4KB
文件格式:ZIP
更新时间:2021-03-27 14:50:51
Python
在PyTorch中将视频加载到Tensor
有很多方法可以播放视频,但是为了提高视频模型的训练速度,我尝试了不同的程序包,看看哪个是最好的。
用法
首先将视频转换为图像(因为一个实验尝试加载帧):
python video2jpg.py
然后,运行实验:
python test_load_video_speed.py
结果
方法
初始数据类型
中间数据类型
最终数据类型
短视频成本/秒
长视频成本/秒
小视频成本/秒
大视频成本/秒
OpenCV
数组
--
张量
0.15
1.08
0.08
15.34
OpenCV
数组
PIL图像
张量
0.27
2.27
0.12
17.35
scikit视频
数组
--
张量
0.43
1.32
0.21
16.00
意象
imageio.core.util.Array
数组
张量
0.71
3.41
0
【文件预览】:
load_video_as_tensor-main
----README.md(2KB)
----video2jpg.py(804B)
----LICENSE(1KB)
----test_load_video_speed.py(11KB)