aleju--imgaug
wehub-resource-sync
37 行
1.1 KiB
Python
37 行
1.1 KiB
Python
from __future__ import print_function, division
|
|
|
|
import numpy as np
|
|
|
|
import imgaug as ia
|
|
import imgaug.augmenters as iaa
|
|
|
|
|
|
def main():
|
|
image = ia.quokka_square((256, 256))
|
|
|
|
reggrid_sampler = iaa.DropoutPointsSampler(
|
|
iaa.RegularGridPointsSampler(n_rows=50, n_cols=50),
|
|
0.5
|
|
)
|
|
uniform_sampler = iaa.UniformPointsSampler(50*50)
|
|
|
|
for p_replace in [1.0, 0.5, 0.1, 0.0]:
|
|
augs = [
|
|
iaa.Voronoi(points_sampler=reggrid_sampler, p_replace=p_replace,
|
|
max_size=128),
|
|
iaa.Voronoi(points_sampler=uniform_sampler, p_replace=p_replace,
|
|
max_size=128),
|
|
iaa.UniformVoronoi(50*50, p_replace=p_replace, max_size=128),
|
|
iaa.RegularGridVoronoi(50, 50, p_drop_points=0.4,
|
|
p_replace=p_replace, max_size=128),
|
|
iaa.RelativeRegularGridVoronoi(p_replace=p_replace, max_size=128)
|
|
]
|
|
|
|
images = [aug(image=image) for aug in augs]
|
|
|
|
ia.imshow(np.hstack(images))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|