多模态吧 关注:78贴子:258
  • 6回复贴,共1
求助

求助pytorch多模态数据集处理

只看楼主收藏回复

我如何从dataloader同时取几个不同的数据集,且保证shuffle是ture,而且这几个数据集取出的数据的标签一样呢?


IP属地:江苏来自Android客户端1楼2024-08-02 08:41回复
    我的那几个数据集的标签都在文件夹上


    IP属地:江苏来自Android客户端2楼2024-08-02 08:42
    回复
      构建一个自定义dataset类继承自dataset,然后在类里定义数据集的len,数据集单个样本的输入以及其标签。多个数据集拼接用dataset concat。shuffle打乱才是在dataloader完成的。


      IP属地:广西来自Android客户端3楼2024-08-07 11:17
      收起回复