[pytorch] 图像分类框架【Efficientnet0/1/2/3/4/5/6/7】

以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 import Dataset,DataLoade

你可能感兴趣的:(比赛方案总结)