无法裁掉透明度,PIL getbbox()和Numpy都无法正常工作

时间:2016-06-21 10:15:00

标签: python image numpy python-imaging-library

在图像处理方面,我非常喜欢菜鸟:( 我有一堆PNG文件(其中300个),我希望裁剪大面积的透明度。我想显然自动化这个过程,因此我尝试使用python和PIL。

现在我来看看以下链接, Crop a PNG image to its minimum size,并且使用此链接建议的Numpy Automatically cropping an image with python/PIL,两者都没有成功:(输出文件与输入文件相同!没有裁剪透明度,大小相同.getbbox是返回相同的宽度和高度。

这是指其中一张图片的链接; 98x50button 图像是钟形按钮图标。它是用白色绘制的,因此很难看出哪个透明背景。预期的结果将是一个20x17的按钮(内部透明度为20x17的盒子保持不变)

这是我使用的代码;

#!/usr/bin/env python

import sys
import os
import Image
import numpy as np


def autocrop_image2(image):
    image.load()

    image_data = np.asarray(image)
    image_data_bw = image_data.max(axis=2)
    non_empty_columns = np.where(image_data_bw.max(axis=0) > 0)[0]
    non_empty_rows = np.where(image_data_bw.max(axis=1) > 0)[0]
    cropBox = (min(non_empty_rows), max(non_empty_rows),
               min(non_empty_columns), max(non_empty_columns))

    image_data_new = image_data[cropBox[0]:cropBox[
        1] + 1, cropBox[2]:cropBox[3] + 1, :]

    new_image = Image.fromarray(image_data_new)
    return new_image


def autocrop_image(image, border=0):
    # Get the bounding box
    bbox = image.getbbox()

    # Crop the image to the contents of the bounding box
    image = image.crop(bbox)

    # Determine the width and height of the cropped image
    (width, height) = image.size

    # Add border
    width += border * 2
    height += border * 2

    # Create a new image object for the output image
    cropped_image = Image.new("RGBA", (width, height), (0, 0, 0, 0))

    # Paste the cropped image onto the new image
    cropped_image.paste(image, (border, border))

    # Done!
    return cropped_image

walk_dir = sys.argv[1]

print('walk_dir = ' + walk_dir)

# If your current working directory may change during script execution, it's recommended to
# immediately convert program arguments to an absolute path. Then the variable root below will
# be an absolute path as well. Example:
# walk_dir = os.path.abspath(walk_dir)
print('walk_dir (absolute) = ' + os.path.abspath(walk_dir))

for root, subdirs, files in os.walk(walk_dir):
    print('--\nroot = ' + root)
    list_file_path = os.path.join(root, 'my-directory-list.txt')
    print('list_file_path = ' + list_file_path)

    with open(list_file_path, 'wb') as list_file:
        for subdir in subdirs:
            print('\t- subdirectory ' + subdir)

        for filename in files:
            file_path = os.path.join(root, filename)

            print('\t- file %s (full path: %s)' % (filename, file_path))
            filename, file_extension = os.path.splitext(filename)
            if file_extension.lower().endswith('.png'):
                # Open the input image
                image = Image.open(file_path)
                # Do the cropping
                # image = autocrop_image(image, 0)
                new_image = autocrop_image2(image)
                # Save the output image
                output = os.path.join("output", filename + ".png")
                print output
                new_image.save(output)

谢谢大家的帮助:)

3 个答案:

答案 0 :(得分:5)

您遇到的问题是您的图片包含透明的白色像素,而您的代码只会剪裁透明黑色的像素。示例图像中大多数像素的RGBA值为(255, 255, 255, 0)

autocrop_image2中,您正在获取频道值的max。您可能只想直接获取alpha通道的值,因此请更改:

image_data_bw = image_data.max(axis=2)

要:

image_data_bw = image_data[:,:,3]

然后该功能的其余部分应按预期工作。

autocrop_image函数存在同样的问题。 getbbox方法返回非零像素的边界,透明白像素不为零。要解决此问题,请尝试将图像从"RGBA"模式转换为预乘alpha "RGBa"模式,然后再找到边界框:

bbox = image.convert("RGBa").getbbox()

答案 1 :(得分:0)

以下是裁剪透明边框的一种解决方案 只需将此脚本与批量.png文件一起放在您的文件夹中:

from PIL import Image
import numpy as np
from os import listdir

def crop(image_name):
    pil_image = Image.open(image_name)
    np_array = np.array(pil_image)
    blank_px = [255, 255, 255, 0]
    mask = np_array != blank_px
    coords = np.argwhere(mask)
    x0, y0, z0 = coords.min(axis=0)
    x1, y1, z1 = coords.max(axis=0) + 1
    cropped_box = np_array[x0:x1, y0:y1, z0:z1]
    pil_image = Image.fromarray(cropped_box, 'RGBA')
    print(pil_image.width, pil_image.height)
    pil_image.save(png_image_name)
    print(png_image_name)

for f in listdir('.'):
    if f.endswith('.png'):
        crop(f)

答案 2 :(得分:0)

这是一个新的解决方案;我只是遇到了这个问题:

您有RGBA图片:

  • 当像素的A为0时,单元格应该完全透明
  • 但是您的某些像素的A为0,RGB值不为零。
  • 枕头的getbbox()和其他函数现在失败。
  • 您希望每当alpha为0时将RGB强制为0

所以:

  • 制作纯黑色RGBA图像,每个像素为(0,0,0,0)
  • 使用您的图像和RGBA黑色图像进行合成 图片作为遮罩。
  • 无论您的A为0还是RGB现在为零
  • 这是一个解决方案;可能有一个内存较低的解决方案。

代码如下:

black = Image.new('RGBA', myImage.size)
myImage = Image.composite(myImage, black, myImage)
myCroppedImage = myImage.crop(myImage.getbbox())