以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
本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。
我来说两句