Random-Erasing-Tensorflow2

Random Erasing Augmentation on Tensorflow2.

This function's purpose is for using map function in tensorflow dataset.

without numpy

Tensorflow version

Tensorflow 2.1.0

Result

There are three result. (probability 0.5)

random erasing(white)

random erasing(black)

random erasing(random)

Test

python main.py

Ref

1. https://arxiv.org/abs/1708.04896, Random Erasing Data Augmentation, Zhun Zhog, Liang Zheng, Guoliang Kang, Shaozi Li, Yi Yang
2. https://github.com/Amitayus/Random-Erasing-TensorFlow.git
3. https://github.com/zhunzhong07/Random-Erasing/blob/master/transforms.py