生成一个自动将图片分类进不同文件夹的代码
时间: 2024-01-28 15:04:33 浏览: 29
好的,以下是一个使用Python的示例代码,可以将指定文件夹中的图片分类到多个文件夹中,分类依据可以是图像的标签或者文件名等:
```python
import os
import shutil
# 指定输入文件夹和输出文件夹
input_folder = 'path/to/input/folder'
output_folder_1 = 'path/to/output/folder1'
output_folder_2 = 'path/to/output/folder2'
# 遍历输入文件夹中的所有文件
for filename in os.listdir(input_folder):
if filename.endswith('.jpg') or filename.endswith('.png'):
# 获取图像标签或文件名等分类依据
label = get_label(filename)
# 根据分类依据将图像复制到对应的输出文件夹中
if label == 'category_1':
shutil.copy(os.path.join(input_folder, filename), output_folder_1)
elif label == 'category_2':
shutil.copy(os.path.join(input_folder, filename), output_folder_2)
else:
# 可以根据实际需求添加更多的分类逻辑
pass
```
其中需要自行实现 `get_label` 函数,该函数接受一个文件名作为输入,返回该文件所属的分类标签。另外,需要将 `input_folder`、`output_folder_1`、`output_folder_2` 等路径替换成实际的文件夹路径。