python-opencv图像扩增操作

import cv2
import numpy as np
import os
img = cv2.imread('E:/pycharm_workspace/python_data_process/pic/fox.jpg')

def rotate_bound(image, angle):
    # grab the dimensions of the image and then determine the
    # center
    (h, w) = image.shape[:2]
    (cX, cY) = (w // 2, h // 2)

    # grab the rotation matrix (applying the negative of the
    # angle to rotate clockwise), then grab the sine and cosine
    # (i.e., the rotation components of the matrix)
    M = cv2.getRotationMatrix2D((cX, cY), -angle, 1.0)
    cos = np.abs(M[0, 0])
    sin = np.abs(M[0, 1])

    # compute the new bounding dimensions of the image
    nW = int((h * sin) + (w * cos))
    nH = int((h * cos) + (w * sin))

    # adjust the rotation matrix to take into account translation
    M[0, 2] += (nW / 2) - cX
    M[1, 2] += (nH / 2) - cY

    # perform the actual rotation and return the image
    return cv2.warpAffine(image, M, (nW, nH))


def process_train_set(data_dir, save_dir, list_file_name):
    file_list = os.path.join(data_dir, list_file_name)

    with open(file_list, "r+") as flist:
        read_data = flist.read()
        flist.truncate()#清空文件
        num=0
        #对单张图像进行处理
        for eachline in read_data.split('\n'):

            img_path, label = os.path.split(eachline)
            img_full_path = os.path.join(data_dir, img_path)
            if(os.path.exists(img_full_path) == False):
                continue
            # 翻转操作（2种）
            img_path_temp = FlipOperation(eachline,label, save_dir)

            # 旋转90和180度
            img_path_temp = Rotation(eachline, label, save_dir)

            #对比度增强
            #直方图正规化
            img_path_temp = Normalize_image(eachline, label, save_dir)

            # 伽马变换
            img_path_temp = gamma_image(eachline, label, save_dir)

            num=num+1
            print("The "+str(num)+" th")
        print("Image argument finished!")

def FlipOperation(eachline,label, save_dir):
    middle_name = 'Flip'
    new_file_path = os.path.join(save_dir, middle_name)
    if not os.path.exists(new_file_path):
        os.makedirs(new_file_path)

    img=cv2.imread(eachline)
    #水平翻转
    flipped1=cv2.flip(img,1)
    cv2.imwrite(new_file_path + "/"+"Hor_" + label,flipped1)

    #垂直翻转
    flipped2 = cv2.flip(img, 0)
    cv2.imwrite(new_file_path + "/" + "Vec_" + label, flipped2)

    return new_file_path

#旋转90和180度
def Rotation(eachline,label, save_dir):
    middle_name = 'Rotation'
    new_file_path = os.path.join(save_dir, middle_name)
    if not os.path.exists(new_file_path):
        os.makedirs(new_file_path)

    img = cv2.imread(eachline)
    rows,cols=img.shape[:2]

    R_image1=cv2.getRotationMatrix2D((rows/2,cols/2),90,1)
    dst = cv2.warpAffine(img, R_image1, (cols, rows))

    dst = cv2.convertScaleAbs(img, alpha=1.5, beta=0.2)

    R_image2 = cv2.getRotationMatrix2D((rows / 2, cols / 2), 180, 1)
    dst1 = cv2.warpAffine(img, R_image2, (cols, rows))
    cv2.imwrite(new_file_path + "/" + "R180_" + label, dst1)
    return new_file_path

#缩放
def Resize(eachline,label, save_dir):
    middle_name = 'Resize'
    new_file_path = os.path.join(save_dir, middle_name)
    if not os.path.exists(new_file_path):
        os.makedirs(new_file_path)

    img = cv2.imread(eachline)
    result=cv2.resize(img,None,fx=2,fy=2,interpolation=cv2.INTER_CUBIC)
    cv2.imwrite(new_file_path + "/" + "2times_" + label, result)
    return new_file_path

#任意角度翻转
def rotate_bound(image, angle):
    # grab the dimensions of the image and then determine the
    # center
    (h, w) = image.shape[:2]
    (cX, cY) = (w // 2, h // 2)

    M = cv2.getRotationMatrix2D((cX, cY), -angle, 1.0)
    cos = np.abs(M[0, 0])
    sin = np.abs(M[0, 1])

    # compute the new bounding dimensions of the image
    nW = int((h * sin) + (w * cos))
    nH = int((h * cos) + (w * sin))

    # adjust the rotation matrix to take into account translation
    M[0, 2] += (nW / 2) - cX
    M[1, 2] += (nH / 2) - cY

    # perform the actual rotation and return the image
    return cv2.warpAffine(image, M, (nW, nH))

#线性增强
def linear_enhance(eachline,label, save_dir):
    middle_name = 'linear'
    new_file_path = os.path.join(save_dir, middle_name)
    if not os.path.exists(new_file_path):
        os.makedirs(new_file_path)

    img = cv2.imread(eachline)
    dst = cv2.convertScaleAbs(img, alpha=1.5, beta=0)
    cv2.imwrite(new_file_path + "/" + label, dst)

#直方图正规化--增强亮度
def Normalize_image(eachline,label, save_dir):
    middle_name = 'normalize'
    new_file_path = os.path.join(save_dir, middle_name)
    if not os.path.exists(new_file_path):
        os.makedirs(new_file_path)

    img = cv2.imread(eachline)
    dst = cv2.normalize(img, dst=None, alpha=350, beta=10, norm_type=cv2.NORM_MINMAX)
    cv2.imwrite(new_file_path + "/" + label, dst)

#伽马变换--能提高对比度
def gamma_image(eachline,label, save_dir):
    middle_name = 'gamma'
    new_file_path = os.path.join(save_dir, middle_name)
    if not os.path.exists(new_file_path):
        os.makedirs(new_file_path)

    img = cv2.imread(eachline)
    img_norm = img / 255.0  # 注意255.0得采用浮点数
    img_gamma = np.power(img_norm, 0.4) * 255.0
    dst = img_gamma.astype(np.uint8)

    cv2.imwrite(new_file_path + "/" + label, dst)

#全局直方图均衡化--灰度图
def EqualizeHist(eachline,label, save_dir):
    middle_name = 'EqualizeHist-gray'
    new_file_path = os.path.join(save_dir, middle_name)
    if not os.path.exists(new_file_path):
        os.makedirs(new_file_path)
    img = cv2.imread(eachline,0)
    dst = cv2.equalizeHist(img)

    cv2.imwrite(new_file_path + "/" + label, dst)

#全局直方图均衡化--彩色图像
def EqualizeHist(eachline,label, save_dir):
    middle_name = 'EqualizeHist-color'
    new_file_path = os.path.join(save_dir, middle_name)
    if not os.path.exists(new_file_path):
        os.makedirs(new_file_path)
    img = cv2.imread(eachline)
    img0 = cv2.equalizeHist(img[:, :, 0])  # 各个通道分别均衡化
    img1 = cv2.equalizeHist(img[:, :, 1])
    img2 = cv2.equalizeHist(img[:, :, 2])
    dst = cv2.merge([img0, img1, img2])
    cv2.imwrite(new_file_path + "/" + label, dst)

data_dir="E:/pycharm_workspace/python_data_process/"
list_file_name ="image.txt"
save_dir="E:/pycharm_workspace/python_data_process/pic_argument1/"
process_train_set(data_dir,save_dir,list_file_name)