数据增强脚本：利用xml标签文件对正样本的目标框位置打上马赛克

用途：

将原有正样本的目标打上码变成负样本

代码（输入输出文件夹路径改为自己的即可）：

import cv2
import xml.etree.ElementTree as ET
import os 

img_path = 'C:\\Users\\hq\\Desktop\\pikaqiu_data\\pikaqiu\\'
xml_path = 'C:\\Users\\hq\\Desktop\\pikaqiu_data\\pikaqiu_xml\\'
mask_img_path = 'C:\\Users\\hq\\Desktop\\pikaqiu_data\\mask_pikaqiu\\'

def xml_mask(img, start, end):
    '''
    depts: 马赛克的大小，图片尺寸小则设置小
    主要通过中间值的rgb对局部范围块的rgb做修改，depts值越小越精确
    '''
    depts = 80
    for i in range(start[0], start[1], depts):
        for j in range(end[0], end[1], depts):
             img[i:min(i + depts, img.shape[0]), j:min(j + depts, img.shape[1])] = \
                    img[(i + min(i + depts, img.shape[0]))//2][(j + min(j + depts, img.shape[1]))//2]
    return img

for img_file in os.listdir(img_path):
    img_filename = os.path.join(img_path, img_file)
    img_cv = cv2.imread(img_filename)
    
    img_name = (os.path.splitext(img_file)[0])
    xml_name = xml_path+'%s.xml'%img_name
    
    root = ET.parse(xml_name).getroot()
    for obj in root.iter('object'):
        name = obj.find('name').text

        xmlbox = obj.find('bndbox')
        x0 = xmlbox.find('xmin').text
        y0 = xmlbox.find('ymin').text
        x1 = xmlbox.find('xmax').text
        y1 = xmlbox.find('ymax').text
        
        xml_mask(img_cv, (int(y0), int(y1)), (int(x0), int(x1)))
        
    cv2.imwrite(mask_img_path + img_file, img_cv)

生成结果：