pythonで画像のbit深度を変換する方法とは

Question

pythonで画像のbit深度を変換する方法を教えていただきたいです。
例えば、CV_8SC3からCV_32FC2に変換するにはどうしたらよいのでしょうか？

featuresという変数をcv2.kmeansにdataとして使った際にCV_32FC2に変換する必要があり、データの型だけ変換したいと考えています。

目的は、画像のクラスタリングをbag of featuresで行うことです。その際にk-meansで、画像の特徴を分類しようとして、エラーが出ました。

コード

retval, bestLabels, codebook=cv2.kmeans(features, self.codebookSize, None, term_crit, iterMax, flags)

エラーコード

TypeError: data data type = 17 is not supported

data type=17がCV_8SC3にあたり、これをCV_32Fまたは、CV_32FC2に変える必要があります。

print(self.codebookSize,term_crit,iterMax,flags)

より、self.codebookSize,term_crit,iterMax,flags = 15 (3, 10, 1) 100 0
となりました。

Comments

[LouiS0616]

タグにOpenCVを追加してください。

[LouiS0616]

また、どのようなルールの下で変換するおつもりなのか追記してください。

[退会済みユーザー]

int8(3ch)からfloat32(2ch)、floatにしてから/255とかだと思ったのですがch数が違うのはどうするんでしょうか...（私も気になります）

[8524ba23]

一般的な色削減でkmeansを使う場合は、チャネル数を変更する必要はないように思えます。やりたいこと＋実際のコードを提示されると回答得られやすいと思います。

[8524ba23]

差支えない範囲で、第三者が再現できるコード、具体的にはself.codebookSize, term_crit, iterMax, flagsの具体的な値が分かるようなコードを追記いただけますか？

Answer 1

元データの縦横２次元画像をreshape((-1,3))で１列にしてnp.float32()すればよいかと思います。
参考：OpenCVのK-Meansクラスタリング

# kmeans
# http://labs.eecs.tottori-u.ac.jp/sd/Member/oyamada/OpenCV/html/py_tutorials/py_ml/py_kmeans/py_kmeans_opencv/py_kmeans_opencv.html
import numpy as np
import cv2

img = cv2.imread('lena.png', cv2.IMREAD_COLOR)
features = img.reshape((-1,3)) # １列に

# convert to np.float32
features = np.float32(features)

# define criteria, number of clusters(K) and apply kmeans()
(codebookSize,term_crit,iterMax,flags) = (15, (3, 10, 1), 100, 0)
retval, bestLabels, codebook=cv2.kmeans(features, codebookSize, None, term_crit, iterMax, flags)

# Now convert back into uint8, and make original image
codebook = np.uint8(codebook)
res = codebook[bestLabels.flatten()]
res2 = res.reshape((img.shape))

cv2.imwrite('res2.png',res2)
cv2.imshow('res2',res2)
cv2.waitKey(0)
cv2.destroyAllWindows()