文件名称:load_video_as_tensor:尝试使用不同的软件包将视频加载到PyTorch中的Tensor中
文件大小:4KB
文件格式:ZIP
更新时间:2024-04-16 08:37:31
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)