以kaggle的SIIM-ISIC Melanoma Classification比賽:https://www.kaggle.com/c/global-wheat-detection ; 介紹efficientnet模型在目標檢測任務上的使用。
數據鏈接:https://www.kaggle.com/c/global-wheat-detection/data
代碼如下: (加入mixup/cutmix)
import sys
sys.path.insert(0, "./timm-efficientdet-pytorch")
sys.path.insert(0, "./omegaconf")
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 albumentations.core.transforms_interface import DualTransform
from albumentations.augmentations import functional as F
from sklearn.model_selection import StratifiedKFold
from torch.utils.data import Dataset,DataLoader
from torch.u