[pytorch] 图像识别之GridMask

MachineLP

发布于 2020-02-18 15:58:39

4.8K0

发布于 2020-02-18 15:58:39

文章被收录于专栏：小鹏的专栏

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 = '../input/bengaliai-cv19'
files_tra

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

如有侵权请联系 cloudcommunity@tencent.com 删除

https