How to use it on your own data set

[Gaoang112]

Hi, thank you for your open source code. Where is the core code of GridMask and how can it be used on your own data set?

[akuxcw]

The codes for ImageNet classification are mainly in :

• https://github.com/Jia-Research-Lab/GridMask/blob/master/imagenet_grid/utils/grid.py
• https://github.com/Jia-Research-Lab/GridMask/blob/master/imagenet_grid/imagenet_amp.py#L134-L136
• https://github.com/Jia-Research-Lab/GridMask/blob/master/imagenet_grid/imagenet_amp.py#L282-L283
• https://github.com/Jia-Research-Lab/GridMask/blob/master/imagenet_grid/imagenet_amp.py#L387-L388

The codes for COCO detection are mainly in :

• https://github.com/Jia-Research-Lab/GridMask/blob/master/detection_grid/maskrcnn_benchmark/data/transforms/grid.py
• https://github.com/Jia-Research-Lab/GridMask/blob/master/detection_grid/train_net.py#L69-L71
• https://github.com/Jia-Research-Lab/GridMask/blob/master/detection_grid/maskrcnn_benchmark/data/transforms/build.py#L44-L45