图像数据增强，最好是用cv2去实现目的

jiachen0212

于 2017-11-27 11:30:59 发布

阅读量4.5k

点赞数

文章标签： cv2 resize rotate

本文链接：https://blog.csdn.net/jiachen0212/article/details/78643188

版权

在做图像数据增强的时候，经常会出现shape错乱的问题而出现很多bug。比较好的做法是，比如resize rotate 这些的函数，统一都使用cv2的，感觉它处理图像还是靠谱的。之前用过transform的rotate resize之类的就比较混乱。。。

给个cv2做rotate的栗子：

import cv2

# 定义旋转rotate函数
def rotate(image, angle, center=None, scale=1.0):
    # 获取图像尺寸
    (h, w) = image.shape[:2]
    # 若未指定旋转中心，则将图像中心设为旋转中心
    if center is None:
        center = (w / 2, h / 2)
    # 执行旋转
    M = cv2.getRotationMatrix2D(center, angle, scale)
    rotated = cv2.warpAffine(image, M, (w, h))
    return rotated
img ＝ img（img，－15）