如何在pytorch中将图像和标签同时提供给DataLoader?

时间:2019-04-01 02:01:20

标签: python pytorch

我有一个2D张量,其中包含我的帧信息,还有一个1D张量,其输出或标签信息与每个帧有关。我做一个函数并返回(框架,标签)。然后,将这两个信息提供给“ torch.utils.data.DataLoader()”。但是DataLoader不能同时用于这两个(框架,标签),而只能为其中之一进行迭代。

0 个答案:

没有答案