GridMask data augmentation implementation in tensorflow

This repo is the implementation of GridMask in pure tensorflow(not official).
Please refer to GridMask for official implementation in pytorch.

Dependencies

python 3.6
Tensorflow 2.1.0

Daisy gridmask before & after

Dog gridmask before & after

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
__pycache__		__pycache__
imgs		imgs
README.md		README.md
gridmask_tf.py		gridmask_tf.py
main.py		main.py
rotate_tf.py		rotate_tf.py