这段代码是一个简单的Python函数,用于将指定文件夹中的图像文件分类到不同的目标文件夹中,具体规则是根据图像的尺寸来分类。让我解释一下代码的功能和如何使用它:
-
导入必要的库:代码中导入了
os
、shutil
和PIL
中的Image
类。这些库用于操作文件系统和图像处理。 -
classify_images
函数:这是主要的函数,它接受三个参数:image_directory
:要分类的图像文件所在的文件夹路径。destination_folder_4k
:用于存放尺寸大于 3000x2000 的图像的目标文件夹路径。destination_folder_1
:用于存放尺寸小于等于 3000x2000 的图像的目标文件夹路径。
-
函数实现:
- 使用
os.listdir(image_directory)
获取指定文件夹中的所有文件名。 - 遍历文件夹中的每个文件。
- 对于以 ".jpg" 或 ".png" 结尾的文件名,打开图像并获取其宽度和高度。
- 根据图像尺寸的大小,决定将图像复制到哪个目标文件夹中。如果图像尺寸大于 3000x2000,则复制到
destination_folder_4k
,否则复制到destination_folder_1
。 - 如果目标文件夹不存在,则创建目标文件夹。
- 使用
shutil.copy2()
函数将图像文件复制到相应的目标文件夹中。
- 使用
-
(重点!!!)使用方法!!!:
- 调用
classify_images
函数在第22行代码!!!,并传入要处理的图像文件夹路径以及两个目标文件夹路径。 - 例如,
classify_images(r"C:\Users\zhish\Downloads", r"C:\Users\zhish\Downloads\4k", r"C:\Users\zhish\Downloads\1")
将会将C:\Users\zhish\Downloads
文件夹中的图像文件分类到C:\Users\zhish\Downloads\4k
或C:\Users\zhish\Downloads\1
文件夹中,具体取决于图像的尺寸。
- 调用