小编jin*_*som的帖子

我目前正在尝试使用堆栈一组图像作为每个标签的单个实体来训练 CNN 使用交叉验证。给定一个 224x224x1 灰度图像的数据集，按以下顺序排序：

Root/
    Class0/image0_view0.png
    Class0/image0_view1.png
    Class0/image0_view2.png
    ...
    Class1/image0_view0.png
    Class1/image0_view1.png
    Class1/image0_view2.png

我将如何将 3 个图像（视图 0、1 和 2）作为尺寸为 224x224x3（3 个灰度图像）的单个张量进行流动？换句话说，我将如何使用 ImageFolder/DatasetFolder 和 DataLoader 在 pytorch 中创建图像堆栈数据集？我是否必须重新组织我的文件夹和类，或者当我进行交叉验证拆分时，制作堆栈会更容易吗？

感谢您的时间和帮助！如果我能提供更多信息，请告诉我。