1. 镜像:
#a:对角镜像,相当于旋转180°
#h:水平镜像
#v:垂直镜像
#镜像
#a:对角镜像,相当于旋转180°
#h:水平镜像
#v:垂直镜像
import cv2
import math
import numpy as np
import os
import glob
import json
import shutil
import xml.etree.ElementTree as ET
from xml.etree.ElementTree import ElementTree, Element
def h_MirrorImg(img_path,img_write_path):
img = cv2.imread(img_path)
mirror_img = cv2.flip(img, 1) #
cv2.imwrite(img_write_path,mirror_img)
def v_MirrorImg(img_path,img_write_path):
img = cv2.imread(img_path)
mirror_img = cv2.flip(img, 0) #
cv2.imwrite(img_write_path,mirror_img)
def a_MirrorImg(img_path,img_write_path):
img = cv2.imread(img_path)
mirror_img = cv2.flip(img, -1) #
cv2.imwrite(img_write_path,mirror_img)
def h_MirrorAnno(anno_path,anno_write_path):
tree = ET.parse(anno_path)
root = tree.getroot()
size=root.find('size')
w=int(size.find('width').text)
objects = root.findall("object")
for obj in objects:
bbox = obj.find('bndbox')
x1 = float(bbox.find('xmin').text)
x2 = float(bbox.find('xmax').text)
x1=w-x1+1
x2=w-x2+1
assert x1>0
assert x2>0
bbox.find('xmin').text=str(int(x2))
bbox.find('xmax').text=str(int(x1))
tree.write(anno_write_path) # 保存修改后的XML文件
def v_MirrorAnno(anno_path,anno_write_path):
tree = ET.parse(anno_path)
root = tree.getroot()
size = root.find('size')
h=int(size.find('height').text)
objects = root.findall("object")
for obj in objects:
bbox = obj.find('bndbox')
y1 = float(bbox.find('ymin').text)
y2 = float(bbox.find('ymax').text)
y1=h-y1+1
y2=h-y2+1
assert y1>0
assert y2>0
bbox.find('ymin').text=str(int(y2))
bbox.find('ymax').text=str(int(y1))
tree.write(anno_write_path) # 保存修改后的XML文件
def a_MirrorAnno(anno_path,anno_write_path):
tree = ET.parse(anno_path)
root = tree.getroot()
size = root.find('size')
w=int(size.find('width').text)
h = int(size.find('height').text)
objects = root.findall("object")
for obj in objects:
bbox = obj.find('bndbox')
x1 = float(bbox.find('xmin').text)
y1 = float(bbox.find('ymin').text)
x2 = float(bbox.find('xmax').text)
y2 = float(bbox.find('ymax').text)
x1=w-x1+1
x2=w-x2+1
y1 = h - y1+1
y2 = h - y2+1
assert x1 > 0
assert x2 > 0
assert y1 > 0
assert y2 > 0
bbox.find('xmin').text=str(int(x2))
bbox.find('xmax').text=str(int(x1))
bbox.find('ymin').text=str(int(y2))
bbox.find('ymax').text=str(int(y1))
tree.write(anno_write_path) # 保存修改后的XML文件
def mirror(img_dir,anno_dir,img_write_dir,anno_write_dir):
if not os.path.exists(img_write_dir):
os.makedirs(img_write_dir)
if not os.path.exists(anno_write_dir):
os.makedirs(anno_write_dir)
img_names=os.listdir(img_dir)
for img_name in img_names:
img_path=os.path.join(img_dir,img_name)
h_img_write_path=os.path.join(img_write_dir,img_name[:-4]+'h'+'.bmp')
anno_path=os.path.join(anno_dir,img_name[:-4]+'.xml')
h_anno_write_path = os.path.join(anno_write_dir, img_name[:-4]+'h'+'.xml')
#
v_img_write_path = os.path.join(img_write_dir, img_name[:-4] + 'v' + '.bmp')
v_anno_write_path = os.path.join(anno_write_dir, img_name[:-4] + 'v' + '.xml')
#
a_img_write_path = os.path.join(img_write_dir, img_name[:-4] + 'a' + '.bmp')
a_anno_write_path = os.path.join(anno_write_dir, img_name[:-4] + 'a' + '.xml')
#
h_MirrorImg(img_path,h_img_write_path)
v_MirrorImg(img_path,v_img_write_path)
a_MirrorImg(img_path, a_img_write_path)
h_MirrorAnno(anno_path,h_anno_write_path)
v_MirrorAnno(anno_path, v_anno_write_path)
a_MirrorAnno(anno_path, a_anno_write_path)
img_dir='C:/Users/123/Desktop/111/1'
anno_dir='C:/Users/123/Desktop/111/2'
img_write_dir='C:/Users/123/Desktop/111/3'
anno_write_dir='C:/Users/123/Desktop/111/4'
mirror(img_dir,anno_dir,img_write_dir,anno_write_dir)
'''
imga=cv2.imread('14a.jpg')
imgh=cv2.imread('14h.jpg')
imgv=cv2.imread('14v.jpg')
cv2.rectangle(imga,(606,898),(855,1432),(0,255,0),4)
cv2.rectangle(imgh,(606,489),(855,1023),(0,255,0),4)
cv2.rectangle(imgv,(32,898),(281,1432),(0,255,0),4)
cv2.imwrite('a.jpg',imga)
cv2.imwrite('v.jpg',imgv)
cv2.imwrite('h.jpg',imgh)
'''
2. 旋转
angle:设置角度
import cv2
import math
import numpy as np
import os
import glob
import json
import shutil
import xml.etree.ElementTree as ET
from xml.etree.ElementTree import ElementTree, Element
def getRotatedImg(Pi_angle,img_path,img_write_path):
img = cv2.imread(img_path)
rows, cols = img.shape[:2]
a, b = cols / 2, rows / 2
M = cv2.getRotationMatrix2D((a, b), angle, 1)
rotated_img = cv2.warpAffine(img, M, (cols, rows)) # 旋转后的图像保持大小不变
cv2.imwrite(img_write_path,rotated_img)
return a,b
def getRotatedAnno(Pi_angle,a,b,anno_path,anno_write_path):
tree = ET.parse(anno_path)
root = tree.getroot()
objects = root.findall("object")
for obj in objects:
bbox = obj.find('bndbox')
x1 = float(bbox.find('xmin').text) - 1
y1 = float(bbox.find('ymin').text) - 1
x2 = float(bbox.find('xmax').text) - 1
y2 = float(bbox.find('ymax').text) - 1
x3=x1
y3=y2
x4=x2
y4=y1
X1 = (x1 - a) * math.cos(Pi_angle) - (y1 - b) * math.sin(Pi_angle) + a
Y1 = (x1 - a) * math.sin(Pi_angle) + (y1 - b) * math.cos(Pi_angle) + b
X2 = (x2 - a) * math.cos(Pi_angle) - (y2 - b) * math.sin(Pi_angle) + a
Y2 = (x2 - a) * math.sin(Pi_angle) + (y2 - b) * math.cos(Pi_angle) + b
X3 = (x3 - a) * math.cos(Pi_angle) - (y3 - b) * math.sin(Pi_angle) + a
Y3 = (x3 - a) * math.sin(Pi_angle) + (y3 - b) * math.cos(Pi_angle) + b
X4 = (x4 - a) * math.cos(Pi_angle) - (y4 - b) * math.sin(Pi_angle) + a
Y4 = (x4 - a) * math.sin(Pi_angle) + (y4 - b) * math.cos(Pi_angle) + b
X_MIN=min(X1,X2,X3,X4)
X_MAX = max(X1, X2, X3, X4)
Y_MIN = min(Y1, Y2, Y3, Y4)
Y_MAX = max(Y1, Y2, Y3, Y4)
bbox.find('xmin').text=str(int(X_MIN))
bbox.find('ymin').text=str(int(Y_MIN))
bbox.find('xmax').text=str(int(X_MAX))
bbox.find('ymax').text=str(int(Y_MAX))
tree.write(anno_write_path) # 保存修改后的XML文件
def rotate(angle,img_dir,anno_dir,img_write_dir,anno_write_dir):
if not os.path.exists(img_write_dir):
os.makedirs(img_write_dir)
if not os.path.exists(anno_write_dir):
os.makedirs(anno_write_dir)
Pi_angle = -angle * math.pi / 180.0 # 弧度制,后面旋转坐标需要用到,注意负号!!!
img_names=os.listdir(img_dir)
for img_name in img_names:
img_path=os.path.join(img_dir,img_name)
img_write_path=os.path.join(img_write_dir,img_name[:-4]+'R'+str(angle)+'.bmp')
#
anno_path=os.path.join(anno_dir,img_name[:-4]+'.xml')
anno_write_path = os.path.join(anno_write_dir, img_name[:-4]+'R'+str(angle)+'.xml')
#
a,b=getRotatedImg(Pi_angle,img_path,img_write_path)
getRotatedAnno(Pi_angle,a,b,anno_path,anno_write_path)
angle=90
img_dir='C:/Users/123/Desktop/111/1'
anno_dir='C:/Users/123/Desktop/111/2'
img_write_dir='C:/Users/123/Desktop/111/3'
anno_write_dir='C:/Users/123/Desktop/111/4'
rotate(angle,img_dir,anno_dir,img_write_dir,anno_write_dir)
3. 对比度
import cv2
import math
import numpy as np
import os
import glob
import json
import shutil
import xml.etree.ElementTree as ET
from xml.etree.ElementTree import ElementTree, Element
def getColorImg(alpha,beta,img_path,img_write_path):
img = cv2.imread(img_path)
colored_img = np.uint8(np.clip((alpha * img + beta), 0, 255))
cv2.imwrite(img_write_path,colored_img)
def getColorAnno(anno_path,anno_write_path):
tree = ET.parse(anno_path)
tree.write(anno_write_path) # 保存修改后的XML文件
def color(alpha,beta,img_dir,anno_dir,img_write_dir,anno_write_dir):
if not os.path.exists(img_write_dir):
os.makedirs(img_write_dir)
if not os.path.exists(anno_write_dir):
os.makedirs(anno_write_dir)
img_names=os.listdir(img_dir)
for img_name in img_names:
img_path=os.path.join(img_dir,img_name)
img_write_path=os.path.join(img_write_dir,img_name[:-4]+'color'+str(int(alpha*10))+'.jpg')
#
anno_path=os.path.join(anno_dir,img_name[:-4]+'.xml')
anno_write_path = os.path.join(anno_write_dir, img_name[:-4]+'color'+str(int(alpha*10))+'.xml')
#
getColorImg(alpha,beta,img_path,img_write_path)
getColorAnno(anno_path,anno_write_path)
alphas=[0.3,0.5,1.2,1.6]
beta=10
img_dir='several/JPEGImages'
anno_dir='several/Annotations'
img_write_dir='Color/color_JPEGImages'
anno_write_dir='Color/color_Annotations'
for alpha in alphas:
color(alpha,beta,img_dir,anno_dir,img_write_dir,anno_write_dir)