web-dev-qa-db-ja.com

Numpy配列からOpenCV配列への変換

私は、白黒画像を表す2D Numpy配列を3チャンネルOpenCV配列（つまり、RGB画像）に変換しようとしています。

コードサンプル および ドキュメント に基づいてPythonのようにこれを実行しようとしています：

import numpy as np, cv
vis = np.zeros((384, 836), np.uint32)
h,w = vis.shape
vis2 = cv.CreateMat(h, w, cv.CV_32FC3)
cv.CvtColor(vis, vis2, cv.CV_GRAY2BGR)

ただし、CvtColor（）を呼び出すと、次のcppレベルの例外がスローされます。

OpenCV Error: Image step is wrong () in cvSetData, file /build/buildd/opencv-2.1.0/src/cxcore/cxarray.cpp, line 902
terminate called after throwing an instance of 'cv::Exception'
  what():  /build/buildd/opencv-2.1.0/src/cxcore/cxarray.cpp:902: error: (-13)  in function cvSetData

Aborted

何が間違っていますか？

pythonnumpyopencvimage-processing
23
Cerin

コードは次のように修正できます。

import numpy as np, cv
vis = np.zeros((384, 836), np.float32)
h,w = vis.shape
vis2 = cv.CreateMat(h, w, cv.CV_32FC3)
vis0 = cv.fromarray(vis)
cv.CvtColor(vis0, vis2, cv.CV_GRAY2BGR)

簡単な説明：

  1. np.uint32データ型はOpenCVでサポートされていません（uint8、int8、uint16、int16、int32、float32をサポートしています、float64）
  2. cv.CvtColorはnumpy配列を処理できないため、両方の引数をOpenCV型に変換する必要があります。 cv.fromarrayはこの変換を行います。
  3. cv.CvtColorの両方の引数は同じ深さでなければなりません。そのため、ソースタイプを宛先に合わせて32ビットのfloatに変更しました。

また、OpenCV python APIの新しいバージョンを使用することをお勧めします。これは、numpy配列をプライマリデータタイプとして使用するためです。

import numpy as np, cv2
vis = np.zeros((384, 836), np.float32)
vis2 = cv2.cvtColor(vis, cv2.COLOR_GRAY2BGR)
29
Andrey Kamaev

これは私のために働いたものです...

import cv2
import numpy as np

#Created an image (really an ndarray) with three channels 
new_image = np.ndarray((3, num_rows, num_cols), dtype=int)

#Did manipulations for my project where my array values went way over 255
#Eventually returned numbers to between 0 and 255

#Converted the datatype to np.uint8
new_image = new_image.astype(np.uint8)

#Separated the channels in my new image
new_image_red, new_image_green, new_image_blue = new_image

#Stacked the channels
new_rgb = np.dstack([new_image_red, new_image_green, new_image_blue])

#Displayed the image
cv2.imshow("WindowNameHere", new_rgbrgb)
cv2.waitKey(0)
2
Mel