线下数据增强:对图像进行上下随机颜色填充,并修改对应的xml文件

本文介绍了线下数据增强的一种方法——对图像进行上下随机颜色填充,同时修改对应的XML文件,确保标注框随图像移动。适用于扩充项目数据,但不适用于颜色识别任务。
摘要由CSDN通过智能技术生成

项目数据不够的时候,往往需要进行数据扩增,此文介绍一种简单的扩增方式,用于线下数据增强。

首先贴上代码

#此脚本用于填充图片,使得图片变成1280*1280大小
import os
import numpy as np
from PIL import Image
from xml.etree.ElementTree import parse, Element

JPEGfilepath = 'JPEGImages'
input_dir = 'JPEGImages/'
out_dir = 'NewJPEGImages/'
a = os.listdir(input_dir)
Annotation_out_dir = 'NewAnnotations/'
b = os.listdir('Annotations/')
for i in a:
    print(i)
    #读取图片的宽高
    image = Image.open(input_dir + i)
    Width = image.size[0]
    Height = image.size[1]
    #设置pad顶部的高
    top = np.random.randint(0, 256)
    #设置pad底部的高
    bottom = 256 - top
    print(top,bottom)
    #设置顶部pad的颜色与需要叠加的顶部图片
    RandColor = (np.random.randint(0,256) , np.random.randint(0,256),np.random.randint(0,256))
    topIm = Image.new('RGB', (1280, top), RandColor)
    # 设置底部pad的颜色与需要叠加的底部图片
    RandColor_2= (np.random.randint(0
  • 4
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值