python - 将图像拆分为任意数量的框

Question

我需要将 RGBA 图像拆分为任意数量的大小尽可能相等的框

我曾尝试使用 numpy.array_split，但不确定如何在保留 RGBA channel 的同时这样做

我看过以下问题，它们都没有详细说明如何将图像拆分为 n 个框，他们提到将图像拆分为预定像素大小的框，或者如何将图像拆分为某种形状。

虽然从盒子大小和图像大小中获取盒子数量似乎是一些简单的数学运算，但我不确定如何去做。

How to Split Image Into Multiple Pieces in Python

Cutting one image into multiple images using the Python Image Library

Divide image into rectangles information in Python

在尝试根据像素框大小确定框数时，我使用了公式

num_boxes = (img_size[0]*img_size[1])/ (box_size_x * box_size_y)

但这并没有导致图像被正确分割

为了澄清，我希望能够输入一个图像，它是一个大小为 (a,b,4) 的 numpy 数组和一些框，并以某种形式输出图像(首选 np 数组，但无论什么都有效)

非常感谢任何帮助，即使您无法提供完整的方法，我也会很感激一些指导。

我试过了

def split_image(image, n_boxes):
    return numpy.array_split(image,n_boxes)
    #doesn't work with colors

def split_image(image, n_boxes):
    box_size = factor_int(n_boxes)
    M = im.shape[0]//box_size[0]
    N = im.shape[1]//box_size[1]

    return [im[x:x+M,y:y+N] for x in range(0,im.shape[0],M) for y in range(0,im.shape[1],N)]

factor_int 从 Factor an integer to something as close to a square as possible 返回尽可能接近正方形的整数

Answer 1

我仍然不确定您的输入实际上是图像和框的尺寸，还是图像和框的数量。我也不确定您的问题是决定在哪里切碎图像还是知道如何切碎 4 channel 图像，但也许这里的内容可以帮助您入门。

我从这张 RGBA 图像开始——圆圈是透明的，不是白色的:

#!/usr/bin/env python3

from PIL import Image
import numpy as np
import math

# Open image and get dimensions
im = Image.open('start.png').convert('RGBA') 

# Make Numpy array from image and get height and width
ni = np.array(im)
h ,w = ni.shape[:2]
print(f'Height: {h}, width: {w}')

BOXES = 4
for i in range(BOXES):
    this = ni[:, i*w//BOXES:(i+1)*w//BOXES, :]
    Image.fromarray(this).save(f'box-{i}.png') 

您可以更改 BOXES 但将其保留在 4 会得到这 4 个输出图像:

[ ] [ ] 4