Pytorch条件数据加载器

时间:2020-05-01 20:38:18

标签: pytorch

比方说,我有一个数据加载器,它返回视频样本,其中每个视频样本都具有以下形状:[number_of_frames X 2048](帧数可以与以下集合不同:[60,80,100,120]。 / p>

我希望从数据加载器返回的每个批次中都具有具有相同帧数的样本。 例如,如果batchsize = 16。我想要以下形状的批次:[16 X 80 X 2048],[16 X 60 X 2048]…
我试图对数据样本进行排序,以使我们将拥有相同长度的连续样本。但是我认为这不是一个好的解决方案。
Anyidea怎么做? 谢谢

0 个答案:

没有答案