tensorflow - 如何从文件夹中加载图像以在 Julia 的 Flux 库中训练 CNN？

Question

我正在尝试在 Julia 的 Flux ML 库中编写一个 CNN（U-Net）。然而，我面临的第一个障碍是，如何从文件夹中加载图像来训练模型。我已经在互联网上广泛搜索了任何方法，但没有运气。互联网上所有关于 Flux 中 CNN 的示例，都使用库函数来导入预先存在的数据集，如 MNIST 数据集。实际上所有的例子，使用一个库函数来加载这个数据集。请有人告诉我，如何从文件夹中加载自定义图像数据集来训练用 Flux Julia 编写的 CNN 模型。数据集的目录结构如下：- 目录结构

Answer 1

图片包 ( https://juliaimages.org/latest/install/ ) 不会自动加载整个图片文件夹。所以这里是加载文件夹内所有图像的代码：

using Images

function load_images(path::String)
    img_paths = readdir(path)
    imgs = []
    for img_path in img_paths
        push!(imgs, load(string(path,img_path)))
    end
    return imgs
end