/Random-Erasing-tensorflow

A complete Tensorflow implementation of cutout random erasing (without numpy)

Primary LanguagePython

Random-Erasing-tensorflow

A data argumentation implementation of random erasing using Tensorflow
WITHOUT numpy !

Note: this side project is just for fun, the performance is equal to numpy version when using i5 CPU, but a bit lower than numpy version when using i7 CPU

Dependency

  1. Tensorflow 1.10+

Result

Test Performance

python3 test_performance.py

Current version test @ CPU i5-3470 & GPU GTX1080ti
numpy: 0.1695 sec/image
TF: 0.1696 sec/image

Ref

  1. Random Erasing Data Augmentation, Zhun Zhong, Liang Zheng, Guoliang Kang, Shaozi Li, Yi Yang
  2. Random Erasing (Author version)
  3. Cutout Random Erasing (Keras version)