-
Notifications
You must be signed in to change notification settings - Fork 0
/
dataAugmentationS.py
79 lines (60 loc) · 2 KB
/
dataAugmentationS.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
# Aumento de dados estático
# Biblioteca
## Gerencia arquivos
from glob import glob
import numpy as np
import pandas as pd
import os
## Transformacao na imagem
import imageio
import imageio as imageio
from PIL import Image
import imgaug as ia
import imgaug.augmenters as iaa
## Gráficos
import matplotlib.pyplot as plt
import matplotlib.patches as patches
import matplotlib
#%matplotlib inline
# Variaveis
i = 0
j = 0
k = 0
l = 0
img_names = glob(os.path.join(os.getcwd(), 'dataset/Dataset_original/forest/*.tiff'))
for imagem in img_names:
#Ler todas as imagens na pasta e plotar
img = imageio.imread(imagem)
# Corta imagens no centro
crop = iaa.CenterCropToFixedSize(height=256, width=256)
crop_img = crop.augment_image(img)
# Salvar imagem
nomeArquivo = os.path.basename(imagem)
Image.fromarray(crop_img).save('dataset/data_DA/forest/'+nomeArquivo)
# Rotação
rotate = iaa.Affine(rotate=(-45, 30))
rotated_image = rotate.augment_image(img)
# Corta imagens no centro
crop = iaa.CenterCropToFixedSize(height=256, width=256)
crop_rotate = crop.augment_image(rotated_image)
# Salvar imagem
nomeArquivo = os.path.basename(imagem)
Image.fromarray(crop_rotate).save('dataset/data_DA/forest/DARot_'+nomeArquivo)
# Flip horizontal
flip_hr = iaa.Fliplr(p=1.0)
flip_hr_image = flip_hr.augment_image(img)
# Corta imagens no centro
crop = iaa.CenterCropToFixedSize(height=256, width=256)
crop_fh = crop.augment_image(img)
# Salvar imagem
nomeArquivo = os.path.basename(imagem)
Image.fromarray(crop_fh).save('dataset/data_DA/forest/DAFlipH_'+nomeArquivo)
# Flip vertical
flip_vr = iaa.Flipud(p=1.0)
flip_vr_image = flip_vr.augment_image(img)
# Corta imagens no centro
crop = iaa.CenterCropToFixedSize(height=256, width=256)
crop_fv = crop.augment_image(flip_vr_image)
# Salvar imagem
nomeArquivo = os.path.basename(imagem)
Image.fromarray(crop_fv).save('dataset/data_DA/savanna/DAFlipV_'+nomeArquivo)