NumPyで画像の分割をしたい

Question

numpyを使用して次の画像を分割して丸を一つ一つにわけて保存したいです。

こちらの記事を参考にコードを書きました。
行方向に４分割しようと思い、以下のコードを書きました。

v_size = img.shape[0] // 4 * 4
h_size = img.shape[1]
img = img[:v_size, :h_size]

v_split = img.shape[0] // 4
out_img = []
out_img.extend(np.vsplit(img,v_split ))

plt.imshow(img)
plt.show()

結果

その結果最後の行と列だけなくなってしまいました。

コードの間違っている点と、参考にした記事のように分割して保存する方法を知りたいです。

↓行・列での分割は以下のようです。

回答よろしくお願いします。

v_size = img.shape[0] // size * size
h_size = img.shape[1] // size * size
img = img[:v_size, :h_size]

v_split = img.shape[0] // size
h_split = img.shape[1] // size
out_img = []
[out_img.extend(np.hsplit(h_img, h_split))
 for h_img in np.vsplit(img, v_split)]

Answer 1

画像の端数を切り捨てる処理で切り捨てられてしまっています。
提示されたコードでは、「分割後のサイズ」から「垂直及び水平方向の分割数」を決めていますが、分割したい数が決まっている場合は、逆に「垂直及び水平方向の分割数」から「分割後のサイズ」を決めたほうがよいでしょう。

import cv2
import matplotlib.pyplot as plt
import numpy as np

img = cv2.imread('colors.jpg')
plt.imshow(img)
plt.show()

h, w, c = img.shape
v_split = 4  # 垂直方向の分割数
h_split = 7  # 水平方向の分割数

# 均等に分割できないと np.spllt() が使えないので、
# 除算したときに余りがでないように画像の端数を切り捨てる。
_img = img[:h // v_split * v_split, :w // h_split * h_split]
print('{} -> {}'.format(img.shape, _img.shape))  # (167, 292, 3) -> (164, 161, 3)

# 分割する。
out_imgs = []
for h_img in np.vsplit(_img, v_split):  # 垂直方向に分割する。
    for v_img in np.hsplit(h_img, h_split):  # 水平方向に分割する。
        out_imgs.append(v_img)
out_imgs = np.array(out_imgs)
print(out_imgs.shape)  # (72, 24, 24, 3)

# 描画する。
fig, axes_list = plt.subplots(v_split, h_split, figsize=(5, 5))
axes_list = axes_list.ravel()
for i, (sub, axes) in enumerate(zip(out_imgs, axes_list)):
    axes.imshow(sub)
    axes.set_axis_off()
    axes.set_title(i)
    
plt.show()

Comments

[24mochimagic]

ありがとうございます！完璧です！