#!/usr/bin/env python
# -*- coding: utf-8 -*-

'''warpAffine实现图像的旋转'''

import cv2
import random

img = cv2.imread("lena.jpg")

# 输入(height,width,channel)，取高度(rows)和宽度(cols)
h,w = img.shape[:2]
# 绕图片中心进行旋转
center = (w / 2, h / 2) 
# 旋转方向取（-180，180）中的随机整数值，逆时针为负，顺时针为正
angle = random.randint(-180, 180)  
# 将图像缩放为80%
scale = 0.8

# 获得旋转矩阵
M = cv2.getRotationMatrix2D(center, angle, scale)
rotated1 = cv2.warpAffine(img,M,(h, w),borderValue=(255, 255, 255))
# rotate函数旋转图像
rotated2 = cv2.rotate(img, cv2.ROTATE_180)

cv2.imshow('img',img)
cv2.imshow('rotated1',rotated1)
cv2.imshow('rotated2',rotated2)
cv2.waitKey(0)
cv2.destroyAllWindows()
 
# 为了输出图片
import sys
sys.path.append('..\\utils') 
import utils
utils.show_img([img,rotated1,rotated2],"fig5-8")