1. ホーム
  2. python

OpenCVによる2枚の画像の合成

2023-09-20 19:24:06

質問

OpenCV 2.1を使って、2つの画像を隣接する1つの画像に結合しようとしています。Pythonで、私はやっています。

import numpy as np, cv

img1 = cv.LoadImage(fn1, 0)
img2 = cv.LoadImage(fn2, 0)

h1, w1 = img1.height,img1.width
h2, w2 = img2.height,img2.width

# Create an array big enough to hold both images next to each other.
vis = np.zeros((max(h1, h2), w1+w2), np.float32)

mat1 = cv.CreateMat(img1.height,img1.width, cv.CV_32FC1)
cv.Convert( img1, mat1 )

mat2 = cv.CreateMat(img2.height, img2.width, cv.CV_32FC1)
cv.Convert( img2, mat2 )

# Copy both images into the composite image.
vis[:h1, :w1] = mat1
vis[:h2, w1:w1+w2] = mat2

h,w = vis.shape
vis2 = cv.CreateMat(h, w, cv.CV_32FC3)
vis0 = cv.fromarray(vis)
cv.CvtColor(vis0, vis2, cv.CV_GRAY2BGR)
cv.ShowImage('test', vis2)
cv.WaitKey()

入力画像は2枚です。

https://code.ros.org/trac/opencv/browser/trunk/opencv/samples/c/box.png?rev=2270

https://code.ros.org/trac/opencv/browser/trunk/opencv/samples/c/box_in_scene.png?rev=2270

出来上がった画像は

他のサイトと見分けがつきにくいかもしれませんが、画像の大部分が白くなっており、個々の画像があるべき場所に対応しています。黒い部分は、画像データが書き込まれていない場所です。

なぜ、画像データがすべて白に変換されてしまうのですか?

どうすればよいのでしょうか?

たまたま画像のサイズが同じだった場合(画像処理結果の表示によくあるケース)、numpyのconcatenateを使ってコードを簡略化することができます。

縦に重ねる場合(img1の上にimg2)。

vis = np.concatenate((img1, img2), axis=0)

水平方向に重ねる(img2の左側にimg1)。

vis = np.concatenate((img1, img2), axis=1)

検証するために

import cv2
import numpy as np
img1 = cv2.imread('img1.png')
img2 = cv2.imread('img2.png')
vis = np.concatenate((img1, img2), axis=1)
cv2.imwrite('out.png', vis)

out.pngの画像は、左がimg1、右がimg2になります。