python实现将指定目录将其目录中的图片放到label中
时间: 2024-11-03 13:19:50 浏览: 33
在Python中,你可以使用`os`模块来遍历文件夹,`glob`模块来匹配特定类型的文件(如图片),以及`shutil`模块来移动文件。这里是一个简单的示例,说明如何将指定目录下的所有图片文件按照它们所在的子目录创建对应的标签文件:
```python
import os
import glob
import shutil
def organize_images(input_dir):
# 指定输入目录
if not os.path.exists(input_dir):
print(f"输入目录 {input_dir} 不存在")
return
# 创建一个dict,键为子目录名,值为该子目录下的图片列表
images_per_folder = {}
# 遍历输入目录及其子目录
for root, dirs, files in os.walk(input_dir):
for folder in dirs:
# 获取子目录下所有的图片文件
img_files = glob.glob(os.path.join(root, folder, '*.{jpg,png,gif}'))
if img_files:
images_per_folder[folder] = img_files
# 现在可以将图片移动到对应的标签文件夹中,这里假设每个子目录就是标签名
for folder, img_list in images_per_folder.items():
if not os.path.exists(folder): # 如果标签目录不存在就创建
os.makedirs(folder)
for img_file in img_list:
shutil.move(img_file, folder) # 移动图片到对应目录
# 使用方法
organize_images('/path/to/input/directory')
```
在这个例子中,你需要替换`'/path/to/input/directory'`为你实际的图片目录路径。运行这个脚本后,它会将输入目录下的所有.jpg、.png和.gif格式的图片移到相应子目录里。
阅读全文