分割した画像を結合したい

Question

前提・実現したいこと

pythonである画像を分割し,画像ごとのHSV値の大小に条件を設けることでそれぞれの画像の色を変えました.
今は9分割しており,色変更後の9枚の画像を結合,1枚の画像にしたいです.
どなたかご教示お願い致します.

発生している問題・エラーメッセージ

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-16-dcc3b2ab874e> in <module>
     96 d.append(pic_show(x) for x in sub_imgs)
     97 img_x = np.vstack((np.hstack(d[0:3]),
---> 98                    np.hstack(d[3:6]),
     99                    np.hstack(d[6:9])
    100                   ))

<__array_function__ internals> in hstack(*args, **kwargs)

~\anaconda3\lib\site-packages\numpy\core\shape_base.py in hstack(tup)
    343         return _nx.concatenate(arrs, 0)
    344     else:
--> 345         return _nx.concatenate(arrs, 1)
    346 
    347 

<__array_function__ internals> in concatenate(*args, **kwargs)

ValueError: need at least one array to concatenate

該当のソースコード

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

def split_imgs(img, hsplits=3, vsplits=3): #画像分割
    h, w = img.shape[:2]
    crop_img = img[: h // vsplits * vsplits, : w // hsplits * hsplits]

    return np.array(
        [x for h_img in np.vsplit(crop_img, vsplits) for x in np.hsplit(h_img, hsplits)]
    )

def pic_show(img): #分割画像を色変更し表示する
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    # チャンネルごとに分解
    h, s, v = cv2.split(hsv)
    mean1 = h.mean()
    mean2 = s.mean()
    mean3 = v.mean()
    
    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
    img_array = np.asarray(img)
    
    if 95 < mean1 < 100:
        img_array[:,:, (1)]=50
    elif 100 < mean1 < 105:
        img_array[:,:, (1)]=100
    elif 105 < mean1 < 110:
        img_array[:,:, (1)]=150
    elif 110 < mean1 < 115:
        img_array[:,:, (1)]=200
    elif 115 < mean1 < 120:
        img_array[:,:, (1)]=250

    plt.imshow(img_array)
    plt.show()

# 画像を読み込む。
hsplits = 3  # 横方向の分割数
vsplits = 3  # 縦方向の分割数
img = cv2.imread("test.jpg")

# 画像を分割する。
sub_imgs = split_imgs(img, hsplits, vsplits)
print(sub_imgs.shape)  

d = []
d.append(pic_show(x) for x in sub_imgs)
img_x = np.vstack((np.hstack(d[0:3]),
                   np.hstack(d[3:6]),
                   np.hstack(d[6:9])
                  ))

cv.imshow(img_x)

試したこと

空のリストを準備して,その中にそれぞれの画像のnp.asarrayを縦横で大きさを指定しながら追加していくことで,
元の画像のように結合できるのでは,と思ったのですが思うように出来ませんでした

補足情報（FW/ツールのバージョンなど）

python 3.7.3

Answer 1

d = []

d.append(pic_show(x) for x in sub_imgs)

ここがうまくリストに追加出来ておりません。
この場合、appendするのではなく直接内包表記自体をリストとして定義しましょう。

d = [pic_show(x) for x in sub_imgs]

しかしこれではpic_show関数を実行しても戻り値が何もない為戻り値も設定します。

def pic_show(img): #分割画像を色変更し表示する
	hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
	# チャンネルごとに分解
	h, s, v = cv2.split(hsv)
	mean1 = h.mean()
	mean2 = s.mean()
	mean3 = v.mean()

	img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
	img_array = np.asarray(img)

	if 95 < mean1 < 100:
		img_array[:,:, (1)]=50
	elif 100 < mean1 < 105:
		img_array[:,:, (1)]=100
	elif 105 < mean1 < 110:
		img_array[:,:, (1)]=150
	elif 110 < mean1 < 115:
		img_array[:,:, (1)]=200
	elif 115 < mean1 < 120:
		img_array[:,:, (1)]=250

	#plt.imshow(img_array)
	#plt.show()
	return img_array

cv.imshow(img_x)

最終行のこちらも色々と惜しいです。cv2としたかったのでしょうか。
いずれも間違いで正しくはpltでしょう。

plt.imshow(img_x)
plt.show()

Comments

[JamesJohn]

ご回答ありがとうございます。
なるほどです,apeend内で繰り返す場合は直接リストとして定義するといいのですね。

また本題と関係なく[None None...None]と表示されており戻り値としてどうすればいいか悩んでおりましたが,それにもお答えいただきありがとうございます。

無事実行・表示させることができました,本当にありがとうございます。
非常に助かりました。

[nto]

内包表記で生成したものはジェネレータとして生成されます。
それらを[]または()で囲ってあげる事でとリストやタプルにする事が出来ます。
今回の場合append内でジェネレータを生成し、そのジェネレータをリスト内に追加するというコードになってしまっておりいわば二次元リストの様な状態になってしまっています。
その為リスト内の要素数は1、その後の処理でエラーが生じてしまいます。
せっかく内包表記での書き方が出来ているので、それを直接[]で囲って変数に代入してあげる事で、正常なリストを作成する事ができるのです。
ちょっと説明下手でごめんなさい。

[JamesJohn]

なるほど、改めて説明して頂いて非常によくわかりました。
本当にありがとうございます。大変勉強になりました。