python实现图片筛选程序

发布时间：2023/12/15

Python实现图片筛选程序的攻略如下：

1. 获取图片

使用Python中的Pillow库可以很容易地获取图片。我们可以使用以下代码来获取指定路径的图片：

from PIL import Image

# 打开图片
im = Image.open('test.jpg')
# 显示图片
im.show()

2. 图片处理

2.1 图片裁剪

我们可以使用Pillow库中的crop函数来裁剪图片。例如，我们想要从图片的(100, 100)处开始裁剪一个大小为200x200的区域，可以使用以下代码：

from PIL import Image

# 打开图片
im = Image.open('test.jpg')
# 裁剪图片
croped_im = im.crop((100, 100, 300, 300))
# 显示裁剪后的图片
croped_im.show()

2.2 图片缩放

我们可以使用Pillow库中的resize函数来缩放图片。例如，我们想要将一张图片的大小缩小一半，可以使用以下代码：

from PIL import Image

# 打开图片
im = Image.open('test.jpg')
# 缩放图片
resized_im = im.resize((im.width//2, im.height//2))
# 显示缩放后的图片
resized_im.show()

3. 图片筛选

为了筛选图片，我们需要定义一个函数，该函数接收一张图片作为参数，然后根据特定的筛选条件判断图片是否符合要求。例如，我们定义一个函数check_image的实现如下：

from PIL import Image

def check_image(im):
    # 获取图片尺寸
    width, height = im.size
    # 判断图片长宽比是否符合要求
    if width/height > 0.5 and width/height < 2:
        return True
    return False

上述函数的作用是判断图片的长宽比是否在0.5到2之间。如果是，则返回True，否则返回False。

4. 完整代码示例

下面是一段完整的代码示例，在这个示例中，我们将获取一张图片，并对这张图片进行裁剪和缩放，最后根据check_image函数对图片进行筛选。

from PIL import Image

def check_image(im):
    # 获取图片尺寸
    width, height = im.size
    # 判断图片长宽比是否符合要求
    if width/height > 0.5 and width/height < 2:
        return True
    return False

# 打开图片
im = Image.open('test.jpg')

# 裁剪图片
croped_im = im.crop((100, 100, 300, 300))
# 显示裁剪后的图片
croped_im.show()

# 缩放图片
resized_im = im.resize((im.width//2, im.height//2))
# 显示缩放后的图片
resized_im.show()

# 筛选图片
if check_image(im):
    print("图片符合要求")
else:
    print("图片不符合要求")

5. 示例说明

示例1：筛选长宽比合适的图片

假设我们需要从一个图片目录中筛选出长宽比在0.5到2之间的图片。我们可以使用以下代码实现：

import os
from PIL import Image

def check_image(im):
    # 获取图片尺寸
    width, height = im.size
    # 判断图片长宽比是否符合要求
    if width/height > 0.5 and width/height < 2:
        return True
    return False

# 指定图片目录
image_dir = '/path/to/image/dir'

# 遍历图片目录
for filename in os.listdir(image_dir):
    # 获取图片路径
    filepath = os.path.join(image_dir, filename)
    # 打开图片
    im = Image.open(filepath)
    # 筛选图片
    if check_image(im):
        print(f"{filename} 符合要求")
    else:
        print(f"{filename} 不符合要求")

示例2：批量缩放图片

假设我们需要批量对一组图片进行缩放，使得它们的大小都小于1000x1000。我们可以使用以下代码实现：

import os
from PIL import Image

# 指定图片目录
image_dir = '/path/to/image/dir'

# 遍历图片目录
for filename in os.listdir(image_dir):
    # 获取图片路径
    filepath = os.path.join(image_dir, filename)
    # 打开图片
    im = Image.open(filepath)
    # 缩放图片
    if im.width > 1000 or im.height > 1000:
        resized_im = im.copy()
        resized_im.thumbnail((1000, 1000), Image.ANTIALIAS)
        # 保存缩放后的图片
        resized_im.save(os.path.join(image_dir, f"resized_{filename}"))

上述代码会将图片缩放成最大不超过1000x1000的大小，并保存到原图片目录下的文件名前面加上"resized_"的文件中。