首页 > 解决方案 > Pytorch 条件数据加载器

问题描述

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

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

标签: pytorch

解决方案


推荐阅读