如何从文件夹加载图像来训练Julia的Flux库中的CNN

Question

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

Answer 1

Images包(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