def textAugment_random(all_data,
Aug_width=False,
Aug_height=False,
Aug_resize=False,
Aug_rotate=False,
Aug_rotate_height=False,
Aug_rotate_width=False,
Aug_add_noise=False,
Aug_change_light=False,
Aug_crop=False,
Aug_shift=False,
iteration=None):
text_aug = ta.TextAugment()
img = readImage(all_data['filepath'])
bboxes = all_data['bboxes']
if Aug_width:
img, bboxes = text_aug.resizeTextWidth(img, bboxes)
if Aug_height:
img, bboxes = text_aug.resizeTextHeight(img, bboxes)
if Aug_resize:
img, bboxes = text_aug.resizeText(img, bboxes)
if Aug_rotate:
img, bboxes = text_aug.rotateText(img, bboxes)
if Aug_rotate_height:
img, bboxes = text_aug.resizeTextHeight(img, bboxes)
if Aug_rotate_width:
img, bboxes = text_aug.resizeTextWidth(img, bboxes)
if Aug_add_noise:
img = text_aug.addNoise(img)
if Aug_change_light:
img = text_aug.changeLight(img)
if Aug_crop:
img, bboxes = text_aug.cropText(img, bboxes)
if Aug_shift:
img, bboxes = text_aug.shiftText(img, bboxes)
tmp_dict = generateAugData(img,
bboxes,
all_data['filepath'],
iteration,
mark='mix')
if ShowPicture:
showPicture(tmp_dict)
def textAugment(all_data,
Aug_width=False,
Aug_height=False,
Aug_resize=False,
Aug_rotate=False,
Aug_rotate_height=False,
Aug_rotate_width=False,
Aug_add_noise=False,
Aug_change_light=False,
Aug_crop=False,
Aug_shift=False,
iteration=None):
text_aug = ta.TextAugment()
if Aug_width:
for i in range(len(all_data)):
img, bboxes = text_aug.resizeTextWidth(
readImage(all_data[i]['filepath']), all_data[i]['bboxes'])
tmp_dict = generateAugData(img,
bboxes,
all_data[i]['filepath'],
iteration,
mark='wid')
if ShowPicture:
showPicture(tmp_dict)
# images.append(tmp_dict)
# image_02 =copy.deepcopy(image_)
# input('stop1')
if Aug_height:
for i in range(len(all_data)):
img, bboxes = text_aug.resizeTextHeight(
readImage(all_data[i]['filepath']), all_data[i]['bboxes'])
tmp_dict = generateAugData(img,
bboxes,
all_data[i]['filepath'],
iteration,
mark='hei')
if ShowPicture:
showPicture(tmp_dict)
# images.append(tmp_dict)
# image_03 =copy.deepcopy(image_02)
if Aug_resize:
for i in range(len(all_data)):
img, bboxes = text_aug.resizeText(
readImage(all_data[i]['filepath']), all_data[i]['bboxes'])
tmp_dict = generateAugData(img,
bboxes,
all_data[i]['filepath'],
iteration,
mark='res')
if ShowPicture:
showPicture(tmp_dict)
# images.append(tmp_dict)
if Aug_rotate:
for i in range(len(all_data)):
img, bboxes = text_aug.rotateText(
readImage(all_data[i]['filepath']), all_data[i]['bboxes'])
tmp_dict = generateAugData(img,
bboxes,
all_data[i]['filepath'],
iteration,
mark='rot')
if ShowPicture:
showPicture(tmp_dict)
# images.append(tmp_dict)
if Aug_rotate_height:
for i in range(len(all_data)):
img, bboxes = text_aug.resizeTextHeight(
readImage(all_data[i]['filepath']), all_data[i]['bboxes'])
img, bboxes = text_aug.rotateText(img, bboxes)
tmp_dict = generateAugData(img,
bboxes,
all_data[i]['filepath'],
iteration,
mark='rhe')
if ShowPicture:
showPicture(tmp_dict)
# images.append(tmp_dict)
if Aug_rotate_width:
for i in range(len(all_data)):
img, bboxes = text_aug.resizeTextWidth(
readImage(all_data[i]['filepath']), all_data[i]['bboxes'])
img, bboxes = text_aug.rotateText(img, bboxes)
tmp_dict = generateAugData(img,
bboxes,
all_data[i]['filepath'],
iteration,
mark='rwi')
if ShowPicture:
showPicture(tmp_dict)
# images.append(tmp_dict)
if Aug_add_noise:
for i in range(len(all_data)):
img = text_aug.addNoise(readImage(all_data[i]['filepath']))
tmp_dict = generateAugData(img,
all_data[i]['bboxes'],
all_data[i]['filepath'],
iteration,
mark='noi')
if ShowPicture:
showPicture(tmp_dict)
# images.append(tmp_dict)
if Aug_change_light:
for i in range(len(all_data)):
img = text_aug.changeLight(readImage(all_data[i]['filepath']))
tmp_dict = generateAugData(img,
all_data[i]['bboxes'],
all_data[i]['filepath'],
iteration,
mark='lig')
if ShowPicture:
showPicture(tmp_dict)
# images.append(tmp_dict)
if Aug_crop:
for i in range(len(all_data)):
img, bboxes = text_aug.cropText(readImage(all_data[i]['filepath']),
all_data[i]['bboxes'])
tmp_dict = generateAugData(img,
bboxes,
all_data[i]['filepath'],
iteration,
mark='crp')
if ShowPicture:
showPicture(tmp_dict)
# images.append(tmp_dict)
if Aug_shift:
for i in range(len(all_data)):
img, bboxes = text_aug.shiftText(
readImage(all_data[i]['filepath']), all_data[i]['bboxes'])
tmp_dict = generateAugData(img,
bboxes,
all_data[i]['filepath'],
iteration,
mark='shf')
if ShowPicture:
showPicture(tmp_dict)