Pythonで画像分割や画像結合を実行する方法

2020年2月25日2021年12月8日

1. 画像分割のやり方
- 1.1. ソースコードのポイント
2. 分割結果（例）
3. 画像結合のやり方
- 3.1. ソースコードのポイント
4. 結合結果（例）
5. まとめ
6. 参考にした書籍
7. 効率的に学習したいなら
- 7.1. おすすめ記事
- 7.2. 関連

画像分割のやり方

以下のプログラムを実行してください。

なお、フォルダ名や画像ファイル名はみなさんの環境に合わせ変更してください。

# 画像分割　sepa.py
import os
import cv2
import numpy as np

# 画像読み込み
image = cv2.imread('./test_photo/001.jpg')

# 画像サイズを調べる（高さ：h 、幅：w）
h, w = image.shape[:2]
n = 3  # 画像分割数
y0 = int(h/n)
x0 = int(w/n) 

# 分割した画像を内包表記でリスト化
c = [image[x0*x:x0*(x+1), y0*y:y0*(y+1)] for x in range(n) for y in range(n)]

# c のリストから1つづつ取り出して
# ファイル番号（0.jpg、1.jpg、・・）を付けて、sepaフォルダに保存
for i, img in enumerate(c):
    cv2.imwrite(os.path.join('./test_photo/sepa', '{}.jpg'.format(i)), img)

cv2を使うので、OpenCVをインストールしていない場合は以下の方法でインストールしてください。

OpenCVのインストール方法

AnacondaがインストールしてあればAnaconda Prompt から以下を実行。

> pip install opencv-python

ソースコードのポイント

1．内包表記で分割画像をリスト化

画像サイズが大きい場合や分割数が多くなると for 文では実行速度が遅くなります。

このため、リスト内包表記を使うのがおすすめです。

2．enumerate関数を使ってインデックスと要素を同時取得

あとあと結合させることを考えファイル番号を数字にしておきます。

これにより一連番号でフォルダにストックされます。

メリットは画像に順番を付けやすいことです。

自然順（ファイル番号の小さい順）に読み込んで結合させれば元通りの画像ができます。

分割結果（例）

画像結合のやり方

以下のプログラムを実行してください。

# 画像結合  comb.py
import cv2
import glob
import numpy as np
from PIL import Image
from natsort import natsorted

# 所定のフォルダ内にある jpg ファイルを連続で読み込んでリスト化する
files = glob.glob("./test_photo/sepa" + "/*.jpg")

# 空のリストを準備
d = []

# natsortedで自然順（ファイル番号の小さい順）に1個づつ読み込む
for i in natsorted(files):
    img = Image.open(i)    # img は'JpegImageFile' object
    img = np.asarray(img)  # np.asarrayで img を ndarray に変換
    d.append(img)          # d にappend で img を追加

# 画像の高さ方向と幅方向を結合
img_x = np.vstack((np.hstack(d[0:3]),
                   np.hstack(d[3:6]),
                   np.hstack(d[6:9])
                  ))

# 色をBGR から RGB に変更
img_x = cv2.cvtColor(img_x, cv2.COLOR_BGR2RGB)
cv2.imshow('img_after', img_x)
cv2.imwrite('./test_photo/result.jpg', img_x)

cv2.waitKey(0)
cv2.destroyAllWindows()