欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

利用Python将图片中扭曲矩形的复原

程序员文章站 2022-06-22 21:21:16
前言日常生活中,手残党们经常会把一些照片拍歪,比如拍个证件、试卷、ppt什么的,比如下面这本书的封面原本是个矩形,随手一拍就成了不规则四边形想要把它变成规则的矩形,有什么办法呢?你一定想到了 ps,因...

前言

日常生活中,手残党们经常会把一些照片拍歪,比如拍个证件、试卷、ppt什么的,

比如下面这本书的封面原本是个矩形,随手一拍就成了不规则四边形

想要把它变成规则的矩形,有什么办法呢?

利用Python将图片中扭曲矩形的复原

你一定想到了 ps,因为网上很多教程就是这么教的

打开ps ⟹\longrightarrow⟹ 滤镜 ⟹\longrightarrow⟹镜头校正 ⟹\longrightarrow⟹自定 ⟹\longrightarrow⟹变换

可是这样手工的调整数值实在是费劲哟!!

利用Python将图片中扭曲矩形的复原

下面我们来写个程序,自动将图片恢复到任意矩形!

程序实现

1. 打开图片并显示

import cv2 as cv
import numpy as np

path = './book.jpg'
src = cv.imread(path)

def get_window_size(src, bound=600):
 h,w = src.shape[0], src.shape[1]
 if h > w:
  h, w = bound, int(w*bound/h)
 else:
  h, w = int(h*bound/w), bound
 return (h, w)

h, w = get_window_size(src)

win_name = 'recadjust'
cv.namedwindow(win_name, cv.window_normal)
cv.resizewindow(win_name, width=w, height=h)
cv.imshow(win_name, src)
cv.waitkey(0)
cv.destroyallwindows()

2. 在图片中标记四个角点

利用Python将图片中扭曲矩形的复原

src_copy = src.copy()

class indexer:
 def __init__(self, bound=4):
  self.id = 0
  self.bound=bound
 
 def get_id(self):
  self.id = (self.id+1)%self.bound
  return (self.id-1)%self.bound+1

def on_event_lbuttondown(event, x, y, flags, param):
 if event == cv.event_lbuttondown:
  
  img = param['src']
  win_name = param['window']
  indexer = param['indexer']
  points = param['points']
  
  curr_id = indexer.get_id()
  points.append((x,y))
  print('第{}个顶点: ({},{})'.format(curr_id, x, y))
  
  cv.circle(img, (x, y), 10, (0, 0, 255), thickness = 2)
  cv.puttext(
   img,  
   str(curr_id),   # 文字
   (x, y),     # 坐标
   cv.font_hershey_plain,
   5,      # 字号
   (0,0,255),    # 字体颜色
   thickness = 2   # 粗细
  )

  cv.imshow(win_name, img)

points = []
indexer = indexer()
win_name = 'recadjust'
cv.namedwindow(win_name, cv.window_normal)
cv.resizewindow(win_name, width=w, height=h)
cv.imshow(win_name, src)
cv.setmousecallback(win_name, on_event_lbuttondown, param={'src':src,'window':win_name, 'indexer':indexer, 'points':points})
cv.waitkey(0)
cv.destroyallwindows()
print(points)

3. 设定输出矩形大小

# w = int(input('输出图片宽度:'))
# h = int(input('输出图片高度:'))
w,h = 600, 800
target_points = [(0,0),(w,0),(w,h),(0,h)]

4. 求解透视变换矩阵

points, target_points = np.array(points, dtype=np.float32), np.array(target_points, dtype=np.float32)
m = cv.getperspectivetransform(points, target_points)
print('透视变换矩阵:',m)

5. 透视变换并保存图片

result = cv.warpperspective(src_copy, m, (0, 0))
result = result[:h, :w]
win_name = 'result'
cv.namedwindow(win_name, cv.window_normal)
cv.resizewindow(win_name, width=w, height=h)
cv.imshow(win_name, result)
cv.waitkey(0)
cv.destroyallwindows()

output_file = 'result.jpg'
cv.imwrite(output_file, result)

结果如下

利用Python将图片中扭曲矩形的复原

结果没有想象的完美,是因为书的封面已经不是一个平面了,书角翘起。。。

如果图片中的矩形在一个平面上的话,效果还是挺不错的!!

利用Python将图片中扭曲矩形的复原

完整代码下载

可关注后免费下载喔:https://download.csdn.net/download/itnerd/12819535(本地下载

总结

到此这篇关于利用python将图片中扭曲矩形复原的文章就介绍到这了,更多相关python图片扭曲矩形复原内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!