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)