[超初心者質問]pythonでの行列作成について

Question

前提・実現したいこと

pythonの初歩的な質問です。
書籍で勉強中なのですが、エラーが出て分かりません。
画像の回転のための行列を作成し、軸反転を行おうとしています。(cv2.flip()を使わずにやろうとしています。）
めちゃくちゃなこと書いてるかもしれませんが、よろしくお願い致します。

該当のソースコード

import numpy as np
import cv2

img = cv2.imread("cleansing_data/sample.jpg")

mat = cv2.getRotationMatrix2D(tuple(np.array([img.size[0]/2, 0])), 180, 1.0)

my_img = cv2.warpAffine(img, mat, img.shape[:2])

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

TypeError Traceback (most recent call last)
<ipython-input-41-9c052fe0ead9> in <module>
4 img = cv2.imread("cleansing_data/sample.jpg")
5
----> 6 mat = cv2.getRotationMatrix2D(tuple(np.array([img.size[0]/2, 0])), 180, 1.0)
7
8 my_img = cv2.warpAffine(img, mat, img.shape[:2])

TypeError: 'int' object is not subscriptable

Comments

[tiitoi]

180度回転させても反転にはならないと思います。

[退会済みユーザー]

最低限何のエラーが発生したのか記載しましょう。

[SnowBerryTart]

中心をx軸中心にして、180度回転させれば実質反転と同じ操作になると考えたのですが、違いましたでしょうか？

[SnowBerryTart]

また反転操作にならなくとも、配列の使い方が間違っていると思っているため、アドバイス頂けると幸いです。

[tiitoi]

> 中心をx軸中心にして、180度回転させれば実質反転と同じ操作になると考えたのですが、違いましたでしょうか？

紙に文字を書いて、180°回転させてみてください。
左右または上下反転とは異なる結果になることがわかると思います。

Answer 1

エラーの原因

img.size[0]/2 が問題。
img.shape[0]/2 と勘違いしている。

解説

TypeError: 'int' object is not subscriptable

型エラー: int オブジェクトは subscriptable ではありません。

subscriptable は Python の用語でオブジェクト obj に対して、obj[0] のような操作ができることを意味します。

terminology - In Python, what does it mean if an object is subscriptable or not? - Stack Overflow

今回、img.size は、subscriptable でない、つまり、img.size[0] のような操作はできないと言われています。

cv2.imread() の返り値 img は ndarray オブジェクトであり、ndarray.size は、その配列の合計バイト数を表す int 型の属性です。画像の大きさ (配列の形状) は size ではなく、shape 属性で取得します。

問題点を修正したコード

import cv2

img = cv2.imread("sample.jpg")
print("img.size", img.size)  # img.size 150975
print("img.shape", img.shape)  # img.shape (183, 275, 3)

h, w, c = img.shape
mat = cv2.getRotationMatrix2D(center=(w // 2, h // 2), angle=180.0, scale=1.0)
dst = cv2.warpAffine(img, mat, dsize=(w, h))
cv2.imwrite('dst.png', dst)

入力

出力

追記

ちなみに画像を左右反転、上下反転させる行列は以下です。

import cv2
import numpy as np

img = cv2.imread("sample.jpg")
h, w, c = img.shape

# 左右反転
invert_x = np.array([[-1.0, 0.0, w],
                     [0.0, 1.0, 0]])
# 上下反転
invert_y = np.array([[1.0, 0.0, 0.0],
                     [0.0, -1.0, h]])

dst = cv2.warpAffine(img, invert_x, dsize=(w, h))
cv2.imwrite("invert_x.png", dst)

dst = cv2.warpAffine(img, invert_y, dsize=(w, h))
cv2.imwrite("invert_y.png", dst)

Comments

[SnowBerryTart]

理解できました！この度は大変わかりやすい回答をご丁寧に解説頂きありがとうございました。