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