[pytorch] 图像识别之GridMask

GridMask: https://arxiv.org/abs/2001.04086

 

albumentations： https://github.com/albumentations-team/albumentations

kaggle的qishenha大佬分享链接：https://www.kaggle.com/c/bengaliai-cv19/discussion/128592。

数据下载：https://www.kaggle.com/c/bengaliai-cv19/data 

代码：

import os
import cv2
import numpy as np
import pandas as pd
import albumentations
from albumentations.core.transforms_interface import DualTransform
from albumentations.augmentations import functional as F
import matplotlib.pyplot as plt
import torch
from torch.utils.data import TensorDataset, DataLoader, Dataset

data_dir = '