将文件中的图片按照某一个标签分类的代码
时间: 2024-03-05 11:53:52 浏览: 99
使用python将图片按标签分入不同文件夹的方法
5星 · 资源好评率100%
以下是一个将文件中的图片按照某一个标签分类的 Python 代码示例:
```python
import shutil
import os
# 定义源文件夹和目标文件夹路径
src_dir = '/path/to/source/folder'
dst_dir = '/path/to/destination/folder'
# 定义标签名称和对应的文件夹名称
label_dict = {'cat': 'cat_images', 'dog': 'dog_images', 'bird': 'bird_images'}
# 遍历源文件夹中的所有文件
for file_name in os.listdir(src_dir):
# 获取文件路径和标签
file_path = os.path.join(src_dir, file_name)
label = file_name.split('_')[0]
# 如果标签不存在于标签字典中,则跳过该文件
if label not in label_dict:
continue
# 获取目标文件夹路径并复制文件
dst_path = os.path.join(dst_dir, label_dict[label])
shutil.copy(file_path, dst_path)
```
上述代码会遍历源文件夹中的所有文件,按照文件名中的标签将文件复制到对应的目标文件夹中。其中,标签名称和对应的文件夹名称存储在 `label_dict` 字典中,可以根据实际需求进行修改。
阅读全文