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

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

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

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

打开PS ⟹\Longrightarrow⟹ 滤镜 ⟹\Longrightarrow⟹镜头校正 ⟹\Longrightarrow⟹自定 ⟹\Longrightarrow⟹变换

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

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

程序实现

1. 打开图片并显示

import cv2 as cv
import numpy as nppath = './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), boundreturn (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 = 0self.bound=bounddef get_id(self):self.id = (self.id+1)%self.boundreturn (self.id-1)%self.bound+1def 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 opencv进行矩形识别
Python opencv进行矩形识别图像识别中,圆形和矩形识别是最常用的两种,上一篇讲解了圆形识别,本例讲解矩形识别,最后的结果是可以识别出圆心,4个顶点,如下图: 左边是原始图像,右边是识别结果 ...
利用OpenCV从图片中提取矩形并标注坐标（室内平面地图）——（一）
某城市会展中心室内地图背景一名室内设计师的日常工作从设计一张会展地图开始.常常有这样的场景:划分除规范的展位后,进入销售阶段,频繁的需要修改这张地图,如展示拆分.合并.换位置.标记已交易. 问题 ...
python代码计算矩形面积_学习资料Python语言基础知识笔记以及答案
01. Python语言基础知识等笔记: 02. Python作业答案: [例2-3]通过输入函数input()输入股票代码.股票名称.当天股票最高价和最低价,通过输出函数print()输出股票代码+ ...
opencv获取图片像素坐标_利用OpenCV从图片中提取矩形并标注坐标（室内平面地图）（一）
某城市会展中心室内地图背景一名室内设计师的日常工作从设计一张会展地图开始.常常有这样的场景:划分除规范的展位后,进入销售阶段,频繁的需要修改这张地图,如展示拆分.合并.换位置.标记已交易. 问题 ...
Python OpenCV 实现魔方识别+复原
简介本项目运用了基础的 Opencv 图像处理算法来实现魔方色块的识别并判断颜色. 用户可以通过将魔方举在摄像机的前方来让机器自动将魔方的色块录入数据,然后系统将会依靠用户录入的数据来产生当前魔方 ...
python 最小外接矩形笔记
目录 opencv生成最小外接矩形: 最小外接矩形修正版: opencv生成最小外接矩形: cnt = np.array([[x1,y1],[x2,y2],[x3,y3],[x4,y4]]) # 必须 ...
Python - 寻找最大矩形数字版矩阵版
一.引言给定一个仅包含 0 和 1 .大小为 rows x cols 的二维二进制矩阵,找出只包含 1 的最大矩形,并返回其面积. 例如给定如上 4 x 5 的 0-1 矩阵,由1组成的最大矩形面积 ...
python定义一个矩形类_创建矩形类
你的class有一些问题.见以下评论class Rectangle: # Init function def __init__(self): # The only members are length ...
python代码计算矩形面积_Python面积计算器源代码
本程序通过一个while循环分两步计算图形面积,首先确认图形形状,然后根据不同的图形采用不同的面积公式进行计算,目前可支持矩形.正方形.三角形.梯形.圆形. 感兴趣的朋友还可以添加其它图形面积计算,也 ...

Python 图片中扭曲矩形的复原