如何从不同的文件夹读取图像数据？

问题描述

我有4个文件夹：猫，狗，老虎和袋鼠，每个文件夹中都保存有100张图像。

当我使用链接器库导入mnist数据集时，会得到图像元组及其相应的标签。我希望以相同的格式阅读并携带我的数据集。

Chainer代码如下：

train,test = chainer.datasets.get_mnist()
train_data,train_targets = np.array(train).transpose()
test_data,test_targets = np.array(test).transpose()
train_data = np.array(list(train_data)).reshape(train_data.shape[0],1,28,28)
test_data = np.array(list(test_data)).reshape(test_data.shape[0],28)

我尝试了下面的代码，但是没有以相同的格式显示代码。

img_dict=dict()

for root,dirs,files in os.walk(path):
    print(os.path.basename(root))
    my_key = os.path.basename(root)

    dir_images = []
    for file_ in files:
        full_file_path = os.path.join(root,file_)
        img = cv2.imread(full_file_path)
        img = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
        dir_images.append(img)

    img_dict[my_key] = dir_images

从链接器库中读取这些图像并将其以与MNIST所述相同的格式带入的正确方法是什么？

解决方法

const {inputMax,inputMin,labelMin,labelMax} = normalizationData;