[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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章