admin管理员组文章数量:1570429
image-perspective-transformation-图片透视变换(投影变换)
文章目录
- image-perspective-transformation-图片透视变换(投影变换)
- 前言:
- 主要功能:
- 参考链接:
- 我的代码链接:
- 核心代码:
- 联系方式:
前言:
最近可能会用到透视变换,做一个数据预处理,因此想起来我之前写过一个透视变换的脚本,今天跑起来,还不够完善,因此将之前夹杂在论文阅读的博客中单独抽取出来,供大家参考~
话不多说,上示例图!
快来一起学习吧!
主要功能:
python处理图片,包括图片平移、图片旋转、图片缩放、图片翻转、透视变换。选择图片中的四个关键点和将要变换的点,用来生成新的透视图
Use translate,scale,flip,rotation to turn one picture into multiple pictures.
参考链接:
https://github/dapsjj/dealAllPicture 主框架用的这位大佬的代码,我加了透视变换和鼠标交互的功能。
我的代码链接:
https://github/kaixindelele/image-perspective-transformation
核心代码:
透视变换函数,首先要获取源区域和目标区域的四个点
def pic_perspective(self,):
...
pts1 = []
pts2 = []
cv2.setMouseCallback("image", self.mouse, param=(image, pts1, pts2))
cv2.waitKey(0)
cv2.destroyAllWindows()
print("pts1:", pts1)
pts1 = np.float32(pts1[:4])
print("pts2:", pts2)
pts2 = np.float32(pts2[:4])
assert len(pts1)==4, "每个只允许四个点"
# 生成透视变换矩阵
M = cv2.getPerspectiveTransform(pts1, pts2)
# 进行透视变换
dst = cv2.warpPerspective(image, M, (image.shape[1], image.shape[0]))
cv2.imwrite('dst.jpg', dst)
# matplotlib默认以RGB通道显示,所以需要用[:, :, ::-1]翻转一下
plt.subplot(121), plt.imshow(image[:, :, ::-1]), plt.title('input')
plt.subplot(122), plt.imshow(dst[:, :, ::-1]), plt.title('output')
plt.show()
# tkinter.messagebox.showinfo('提示', '透视变换的图片处理完毕!')
return dst
对鼠标的坐标点获取:
def mouse(self, event, x, y, flags, param):
image = param[0]
pts1 = param[1]
pts2 = param[2]
# 如果摁住左键,将坐标赋值给pts1
if event == cv2.EVENT_LBUTTONDOWN:
pts1.append([x, y])
xy = "%d,%d" % (x, y)
cv2.circle(image, (x, y), 4, (0, 255, 255), thickness = -1)
cv2.putText(image, xy, (x, y), cv2.FONT_HERSHEY_PLAIN,
1.0, (0, 255, 255), thickness = 2)
cv2.imshow("image", image)
# 如果摁住右键,将坐标赋值给pts2
if event == cv2.EVENT_RBUTTONDOWN:
pts2.append([x, y])
xy = "%d,%d" % (x, y)
cv2.circle(image, (x, y), 4, (255, 0, 255), thickness = -1)
cv2.putText(image, xy, (x, y), cv2.FONT_HERSHEY_PLAIN,
1.0, (255, 0, 255), thickness = 2)
cv2.imshow("image", image)
联系方式:
ps: 欢迎做强化的同学加群一起学习:
深度强化学习-DRL:799378128
欢迎关注知乎帐号:未入门的炼丹学徒
CSDN帐号:https://blog.csdn/hehedadaq
极简spinup+HER+PER代码实现:https://github/kaixindelele/DRLib
本文标签: 透视图片Imageperspectivetransformation
版权声明:本文标题:image-perspective-transformation-图片透视变换(投影变换) 内容由热心网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:https://www.elefans.com/dongtai/1725894541a1047470.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论