前言
python + opencv实现图像矫正,并解决中文文件名报错
一、实现效果
矫正前效果:
矫正后效果:
二、代码
1.引入库
代码如下(示例):
import cv2
import numpy as np
def pic(img_url, lt_x, lt_y, rt_x, rt_y, lb_x, lb_y, rb_x, rb_y, img_width, img_height):
"""
图像通过透视变换矫正
:param img_url: 图片路径
:param lt_x: 左上x
:param lt_y: 左上y
:param rt_x: 右上x
:param rt_y: 右下y
:param lb_x: 左下x
:param lb_y: 左下y
:param rb_x: 右下x
:param rb_y: 右下y
:param img_width: 图片宽
:param img_height: 图片高
:return:
"""
# 解决imread()读取中文名字的图片报错
img = cv2.imdecode(np.fromfile(img_url, dtype=np.uint8), -1)
# img = cv2.imread(img_url)
row, cols, _ = img.shape
pts_o = np.float32([[lt_x, lt_y], [rt_x, rt_y], [lb_x, lb_y], [rb_x, rb_y]])
pts_d = np.float32([[0, 0], [img_width, 0], [0, img_height], [img_width, img_height]])
m = cv2.getPerspectiveTransform(pts_o, pts_d)
dst = cv2.warpPerspective(img, m, (img_width, img_height))
# 解决imwrite()保存含有中文路径乱码
cv2.imencode('.png', dst)[1].tofile('(图片保存路径)\\qiaoba_副本1.png')
# cv2.imwrite('(图片保存路径)\\qiaoba_副本1.png', dst)
# cv2.imshow('dst', dst)
# # # 等待键盘输入0
# cv2.waitKey(0)
# # # 摧毁所有窗口
# cv2.destroyAl1Windows()
if __name__ == '__main__':
pic('(图片路径)\\qiaoba_副本.png', 71.5, 112, 298.5, 113, 53.5, 573, 306.5, 624, 600, 800)