以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 cv