我要评论前言
日常生活中，手残党们经常会把一些照片拍歪，比如拍个证件、试卷、ppt什么的，
比如下面这本书的封面原本是个矩形，随手一拍就成了不规则四边形
想要把它变成规则的矩形，有什么办法呢？
你一定想到了 ps，因为网上很多教程就是这么教的
打开ps
可是这样手工的调整数值实在是费劲哟！！
下面我们来写个程序，自动将图片恢复到任意矩形！
程序实现
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. 在图片中标记四个角点
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)
结果如下
结果没有想象的完美，是因为书的封面已经不是一个平面了，书角翘起。。。
如果图片中的矩形在一个平面上的话，效果还是挺不错的！！
完整代码下载
可关注后免费下载喔：https://download.csdn.net/download/itnerd/12819535（）
总结
到此这篇关于利用python将图片中扭曲矩形复原的文章就介绍到这了,更多相关python图片扭曲矩形复原内容请搜索移动技术网以前的文章或继续浏览下面的相关文章希望大家以后多多支持移动技术网！
您可能感兴趣的文章:
如您对本文有疑问或者有任何想说的，请点击进行留言回复，万千网友为您解惑！
网友评论