画像のゆがみ補正

まだオートでの補正は出来ていませんが

取り敢えず元画像と結果画像をアップします

元画像

補正画像

コード

import cv2
import numpy as np
from matplotlib import pyplot as plt

path = 'yugami.jpg' # 画像のパス
i = cv2.imread(path, 1) # 画像読み込み
print(i.shape)

# 変換前後の対応点を設定
p_original = np.float32([[670,1050], [2300,1100], [205, 3450], [2580, 3530]])
p_trans = np.float32([[0,0], [3000,0], [0,4000], [3000,4000]])

# 変換マトリクスと射影変換
M = cv2.getPerspectiveTransform(p_original, p_trans)
i_trans = cv2.warpPerspective(i, M, (3000, 4000))

cv2.imwrite("out.jpg", i_trans)

#ここからグラフ設定
fig = plt.figure()
ax1 = fig.add_subplot(111)

# 画像をプロット
show = cv2.cvtColor(i_trans, cv2.COLOR_BGR2RGB)
ax1.imshow(show)

fig.tight_layout()
plt.show()
plt.close()

歪み補正