以kaggle的SIIM-ISIC Melanoma Classification比赛:https://www.kaggle.com/c/siim-isic-melanoma-classification ; 介绍efficientnet模型在分类任务上的使用。
数据链接:https://www.kaggle.com/shonenkov/melanoma-merged-external-data-512x512-jpeg (融合2019和2010年数据)
代码如下: (加入mixup/cutmix)
from glob import glob
import pandas as pd
from sklearn.model_selection import GroupKFold
import cv2
from skimage import io
import albumentations as A
import torch
import os
from datetime import datetime
import time
import random
import cv2
import pandas as pd
import numpy as np
import albumentations as A
import matplotlib.pyplot as plt
from albumentations.pytorch.transforms import ToTensorV2
from sklearn.model_selection import StratifiedKFold
from torch.utils.data