将图像字典保存到本地机器

Question

我有一本图片和类标签字典：

{
    'dog': ['dog1.jpg', 'dog2.jpg', 'dog3.jpg', ...],
    'cat': ['cat1.jpg', 'cat2.jpg', ...]
}

如何根据字典键将这些本地保存在windows机器上的文件夹中？

也就是说，我有一个包含所有狗图像的“狗”文件夹，一个包含所有猫图像的“猫”文件夹等等。

Answer 1

这个对你有用吗？

import os

data = {
    "dog": ["dog1.jpg", "dog2.jpg", "dog3.jpg"],
    "cat": ["cat1.jpg", "cat2.jpg"]
}

for folder_name, images in data.items():
    dest_path = f"some_path/{folder_name}"
    os.mkdir(dest_path)
    for image in images:
        os.rename(f"some_path2/{image}", f"{dest_path}/{image}")

我假设您在某个目录中有所有的图像，并希望通过将狗图像放到狗目录中来将它们分开，等等。