Perspective Transform

lib opencv, numpy, pillow

 

main.py :: 특정 파일을 기준으로 이미지 변환

user_select.py :: 사용자의 윈도우 화면 캡쳐 후 이미지 변환

 

import cv2, os
import numpy as np
from PIL import ImageGrab

filename = "capture"
ext = ".jpg"


img_window = ImageGrab.grab()
img_np = np.array(img_window)
frame = cv2.cvtColor(img_np, cv2.IMREAD_COLOR)
ori_img = frame

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:
      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=(width, height))

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

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

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

full source

github https://github.com/DevDiabloH/PerspectiveTransform_Python

DevDiabloH/PerspectiveTransform_Python