目标检测中的数据增强方法

在计算机视觉方面，计算机视觉的主要问题是没有办法得到充足的数据。对大多数机器学习应用，这不是问题，但是对计算机视觉，数据就远远不够。所以这就意味着当你训练计算机视觉模型的时候，数据增强会有所帮助，这是可行的，无论你是使用迁移学习，使用别人的预训练模型开始，或者从源代码开始训练模型。

下面就详细介绍数据增强方式以及代码实现讲解。

数据增强

数据增强可以使得我们的图片变得更加多样，比如你的数据集中只有左边的这样的一张图片,假如我们分别对其进行色调和翻转等操作，就可以得到四张新的不同含有猫的照片，这样你的数据集就扩充了，猫的识别效果会更好，因为数据集样本更加的多样性，放入神经网络训练可以提高网络的鲁棒性，降低各方面的额外因素对识别的影响，比如：亮度和形态，你的数据集已经包括不同形态猫的图片。

通过改变亮度，图像扭曲等方式使得图像变得更加多种多样，如上图所示，尽管亮度，形态发生了细微改变，但本质上图片上的目标还是一只猫。

目标检测中的数据增强

如果你只是想简单的进行目标分类任务，让你的网络模型遍历图片，判断这是不是一只猫，像上面那样进行数据扩充之后保存图片到数据集中再喂入到网络中训练即可，但是在目标检测中不仅需要进行目标分类还要进行目标定位，这就涉及到目标框的调整了。

一般的数据增强方式：

对图像进行缩放并进行长和宽的扭曲

对图像进行翻转

对图像进行色域扭曲

在目标检测中，并不是直接增强图片就好了，我们要考虑扭曲以及翻转之后框的位置，也就是说框的位置要随着图片的改变而改变。

右图是我用VOC2007数据集中的一张图片，包含人和狗两个目标以及相应的框框，左图是我们进行数据增强之后得到的一张新的图片，我们可以看到我们的原图的宽高被扭曲、尺寸被缩小了、图片被翻转、图片的色域也发生了变化，但是目标以及框住目标的框框依然存在。

实现代码

使用以下代码，即可实现对VOC2007的数据集进行数据增强，代码的讲解已经详细注释好了。

fromPILimportImage,ImageDraw

importnumpyasnp

frommatplotlib.colorsimportrgb_to_hsv,hsv_to_rgb

"""

数据增强的方式:

数据增强其实就是让图片变得更加多样,数据增强是非常重要的提高目标检测算法鲁棒性的手段。

可以通过改变亮度，图像扭曲等方式使得图像变得更加多种多样，改变后的图片放入神经网络进行训练可以提高网络的鲁棒性，降低各方面额外因素对识别的影响.

"""

random preprocessing for real-time data augmentation

:paramannotation_line:数据集中的某一行对应的图片

:paraminput_shape: yolo网络输入图片的大小416*416

:paramrandom:

:parammax_boxes:

:paramjitter:控制图片的宽高的扭曲比率,jitter=.5表示在0.5到1.5之间进行扭曲

:paramhue:代表hsv色域中三个通道中的色调进行扭曲，色调（H）=.1

:paramsat:代表hsv色域中三个通道中的饱和度进行扭曲，饱和度(S)=1.5

:paramval:代表hsv色域中三个通道中的明度进行扭曲，明度（V）=1.5

:paramproc_img:

:return:

"""

line = annotation_line.split()

image = Image.open(line[])

iw,ih = image.size#原图片大小

h,w = input_shape#模型输入图片的大小

box = np.array([np.array(list(map(int,box.split(','))))forboxinline[1:]])#对该行的图片中的目标框进行一个划分

#对图像进行缩放并且进行长和宽的扭曲

#扭曲后的图片大小可能会大于416*416的大小，但是在加灰条的时候会修正为416*416

new_ar = w/h * rand(1-jitter,1+jitter)/rand(1-jitter,1+jitter)#表原图片的宽高的扭曲比率，jitter=0,则原图的宽高的比率不变，否则对图片的宽和高进行一定的扭曲

scale = rand(.25,2)# scale控制对原图片的缩放比率，rand(.25, 2)表示在0.25到2之间缩放，图片可能会放大可能会缩小，rand(.25, 1)会把原始的图片进行缩小，图片的边缘加上灰条，可以训练网络对我们小目标的检测能力。rand(1,2)则是一定放大图像

ifnew_ar

nh =int(scale * h)

nw =int(nh * new_ar)

else:

nw =int(scale * w)

nh =int(nw / new_ar)

image = image.resize((nw,nh),Image.BICUBIC)

# print(nw,nh) #扭曲后的图片的宽和高

#将图像多余的部分加上灰条，一定保证图片的大小为w,h = 416,416

dx =int(rand(,w - nw))

dy =int(rand(,h - nh))

new_image = Image.new('RGB',(w,h),(128,128,128))# (128, 128, 128)代表灰色

new_image.paste(image,(dx,dy))

image = new_image

#是否翻转图像

flip = rand()

ifflip: image = image.transpose(Image.FLIP_LEFT_RIGHT)#左右翻转

#色域扭曲

#色域扭曲是发生在这个hsv这样的色域上，hsv色域是有色调H、饱和度S、明度V三者控制，调整这3个值调整色域扭曲的比率

hue = rand(-hue,hue)

sat = rand(1,sat)ifrand()

val = rand(1,val)ifrand()

x = rgb_to_hsv(np.array(image) /255.)#将图片从RGB图像调整到hsv色域上之后，再对其色域进行扭曲

x[...,] += hue

x[...,][x[...,] >1] -=1

x[...,][x[...,]

x[...,1] *= sat

x[...,2] *= val

x[x >1] =1

x[x

image_data = hsv_to_rgb(x)# numpy array, 0 to 1

#将box进行调整

#对原图片进项扭曲后，也要对原图片中的框框也进行相应的调整

box_data = np.zeros((max_boxes,5))

iflen(box) >:

np.random.shuffle(box)

#扭曲调整

box[:,[,2]] = box[:,[,2]] * nw / iw + dx

box[:,[1,3]] = box[:,[1,3]] * nh / ih + dy

#旋转调整

ifflip: box[:,[,2]] = w - box[:,[2,]]

#因为调整后不再图像中的目标框的调整

box[:,:2][box[:,:2]

box[:,2][box[:,2] > w] = w

box[:,3][box[:,3] > h] = h

box_w = box[:,2] - box[:,]

box_h = box[:,3] - box[:,1]

box = box[np.logical_and(box_w >1,box_h >1)]# discard invalid box

iflen(box) > max_boxes: box = box[:max_boxes]

box_data[:len(box)] = box

returnimage_data,box_data

#原图片绘制展示

defnormal_(annotation_line,input_shape):

"""

random preprocessing for real-time data augmentation

:paramannotation_line:选取的数据集第a行所对应的图片进行数据增强

:paraminput_shape:输入的大小

:return:

"""

line = annotation_line.split()#以空格进行分割

#获取该行对应的图片

image = Image.open(line[])

#获取该图片上的每一个目标框

box = np.array([np.array(list(map(int,box.split(','))))forboxinline[1:]])

returnimage,box

if__name__ =="__main__":

withopen("2007_train.txt")asf:

lines = f.readlines()

a = np.random.randint(,len(lines))

line = lines[a]#选取的数据集第a行所对应的图片进行数据增强

image_data,box_data = normal_(line,[416,416])

img = image_data

#原图片绘制展示

#数据集的第a行图片的展现

forjinrange(len(box_data)):

thickness =3

left,top,right,bottom = box_data[j][:4]

draw = ImageDraw.Draw(img)

foriinrange(thickness):

draw.rectangle([left + i,top + i,right - i,bottom - i],outline=(255,255,255))

img.show()

#对图片进行数据增强后的展示

image_data,box_data = get_random_data(line,[416,416])

print(box_data)

img = Image.fromarray((image_data *255).astype(np.uint8))

forjinrange(len(box_data)):

thickness =3

left,top,right,bottom = box_data[j][:4]

#创建绘制对象

draw = ImageDraw.Draw(img)

foriinrange(thickness):

draw.rectangle([left + i,top + i,right - i,bottom - i],outline=(255,255,255))

img.show()

从代码中我们可以看到的主要步骤是：读取数据集图像－＞对图像做数据做数据增强－＞调整图像中的目标框。