2次元配列の和で、3次元配列（カラー画像)の作成がしたい

Question

列1	列2	列3
1	1	1
0	0	0
0	0	0

列1	列2	列3
0	0	0
1	1	1
0	0	0

上記の2つの2次元配列を1つの3次元配列にまとめて、カラー画像を作成したいです。

以下のプログラムだと2次元配列のままで、画像として保存できなく困っております。
何か助言がある方、お願いいたします。

import cv2
import pandas as pd
import matplotlib.pyplot as plt

data1 = pd.read_csv('1.csv', header = None)
data2 = pd.read_csv('2.csv', header = None)

data1_np = data1.values
data2_np = data2.values

data_sum = data1_np + data2_np

cv2.imwrite('data.png, data_sum)

Answer 1

まず、画像を保存できるために正しいチャンネル数の行列（多次元配列）を用意する必要があります。
普段の画像は1チャンネル（grayscale)か、3チャンネル（RGB)か、4チャンネル（RGBA、赤緑青と透明感の４チャンネル）です。

３つの２次元配列があれば、NumPyを使ってnp.stackという関数でまとめることができます。(x,y)の形の配列から(3,x,y)の形の配列を作ります。

OpenCVで保存するときに、配列を回転する必要があります。３チャンネル画像の場合は、先ほど作った(3,x,y)の配列のaxisを変えて、(x,y,3)の形に変換します。最初から色チャンネルが最後のaxisという配列があれば、もちろん回転する必要がないです。

ちなみに、OpenCVの色のチャンネルの順番はRGBじゃなくてBGRです。

以下のコードで3チャンネルのランダムノイズの5x5の画像が作られます。画像がもっと見えやすくなるようにa,b,c,のsizeを大きくしてみてください。

import numpy as np
import cv2

a = np.random.randint(0, 256, size=(5, 5))
b = np.random.randint(0, 256, size=(5, 5))
c = np.random.randint(0, 256, size=(5, 5))

d = np.stack((a, b, c))

print("A:")
print(a)
print("B:")
print(b)
print("C:")
print(c)
print("D:")
print(d)
print(d.shape)

d = np.transpose(d, (1, 2, 0))
print(d.shape)

cv2.imwrite('temp.png', d)

Comments

[masukin]

ありがとうございます！思った通りに出力することができました。
大変助かりました。

Answer 2

https://deepage.net/features/numpy-stack.html

を参考にされてみてはどうでしょうか。
この記事でわからない場合は、記事内に出てくる関数を調べていくと良いと思います。

Comments

[masukin]

回答ありがとうございます。