이미지펼치기

Github Source Link
출처 : https://opentutorials.org/module/3811

모듈 import

import cv2, os
import numpy as np

원본이미지 불러오기

img_path = './images/samples/pexels-photo-276690.jpg'
filename, ext = os.path.splitext(os.path.basename(img_path))
ori_img = cv2.imread(img_path)
src = []  # 마우스 좌표값 입력

이미지 펼치기

# mouse callback handler
def mouse_handler(event, x, y, flags, param):
    if event == cv2.EVENT_LBUTTONUP:
        img = ori_img.copy()

        src.append([x, y])

        for xx, yy in src:  # 각각의 클릭지점에 점을 생성
            cv2.circle(img, center=(xx, yy), radius=5, color=(0, 255, 0), thickness=-1, lineType=cv2.LINE_AA)

        cv2.imshow('img', img)  # 점이 찍힌 이미지 디스플레이

        # perspective transform
        if len(src) == 4:  # 4개의 점이 모두 찍혔으면
            src_np = np.array(src, dtype=np.float32)

            width = max(np.linalg.norm(src_np[0] - src_np[1]), np.linalg.norm(src_np[2] - src_np[3]))
            height = max(np.linalg.norm(src_np[0] - src_np[3]), np.linalg.norm(src_np[1] - src_np[2]))

            dst_np = np.array([
                [0, 0],
                [width, 0],
                [width, height],
                [0, height]
            ], dtype=np.float32)

            M = cv2.getPerspectiveTransform(src=src_np, dst=dst_np)

            result = cv2.warpPerspective(ori_img, M=M, dsize=(int(width), int(height)))

            cv2.imshow('result', result)
            cv2.imwrite('./images/results/%s_result%s' % (filename, ext), result)

원본이미지를 디스플레이 하고 마우스 콜백 함수 입력

# main
cv2.namedWindow('img')
cv2.setMouseCallback('img', mouse_handler)

cv2.imshow('img', ori_img)
cv2.waitKey(0)