|
|
import cv2
|
|
|
import sys
|
|
|
import os
|
|
|
import numpy as np
|
|
|
#仿射变化
|
|
|
def Affinevariation():
|
|
|
cap = cv2.VideoCapture(0)
|
|
|
ok, src = cap.read()
|
|
|
cap.release()
|
|
|
if ok:
|
|
|
# 图像放缩
|
|
|
src = cv2.resize(src, (256, 256))
|
|
|
# 获取图像shape
|
|
|
rows, cols = src.shape[: 2]
|
|
|
########Begin########
|
|
|
#设置图像仿射变化矩阵
|
|
|
post1 = np.float32([[50, 50], [200, 50], [50, 200]])
|
|
|
post2 = np.float32([[10, 100], [200, 50], [100, 250]])
|
|
|
M = cv2.getAffineTransform(post1, post2)
|
|
|
# 图像仿射变换
|
|
|
result = cv2.warpAffine(src, M, (rows, cols))
|
|
|
img = np.hstack([src, result])
|
|
|
cv2.imshow("resourse and result", img)
|
|
|
# 等待关闭
|
|
|
cv2.waitKey(0)
|
|
|
#图形扩展缩放、平移、旋转
|
|
|
def Graphics_extend_scale_pan_rotate():
|
|
|
cap = cv2.VideoCapture(0)
|
|
|
ok, src = cap.read()
|
|
|
cap.release()
|
|
|
if ok:
|
|
|
# 图像放缩
|
|
|
img = src
|
|
|
l, w, h = img.shape
|
|
|
# 放大图像至原来的两倍，使用双线性插值法
|
|
|
|
|
|
cv2.resize(img, (0, 0), 2, 2, cv2.INTER_LINEAR)
|
|
|
|
|
|
height, width, channel = img.shape
|
|
|
# 构建移动矩阵,x轴左移 10 个像素，y轴下移 30 个
|
|
|
|
|
|
M = np.float32([[1, 0, 10], [0, 1, 30]])
|
|
|
|
|
|
img = cv2.warpAffine(img, M, (width, height))
|
|
|
# 构建矩阵，旋转中心坐标为处理后图片长宽的一半，旋转角度为45度，缩放因子为1
|
|
|
|
|
|
M = cv2.getRotationMatrix2D((width / 2, height / 2), 45, 1)
|
|
|
|
|
|
dst = cv2.warpAffine(img, M, (width, height))
|
|
|
|
|
|
img = np.hstack([src, dst])
|
|
|
|
|
|
cv2.imshow("resourse and result", img)
|
|
|
# 等待关闭
|
|
|
cv2.waitKey(0)
|
|
|
def main():
|
|
|
print("1.仿射变化 2.图形扩展缩放、平移、旋转")
|
|
|
myinput = input()
|
|
|
if myinput == '1':
|
|
|
Affinevariation()
|
|
|
elif myinput == '2':
|
|
|
Graphics_extend_scale_pan_rotate()
|
|
|
else:
|
|
|
print("wrong input!")
|
|
|
|
|
|
|
