README.md

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

Example

Daisy gridmask before & after

Dog gridmask before & after

Reference

GridMask paper
Official GridMask repo