前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >[pytorch] 图像目标检测框架【Efficientnet0/1/2/3/4/5/6/7】

[pytorch] 图像目标检测框架【Efficientnet0/1/2/3/4/5/6/7】

作者头像
MachineLP
发布2020-10-29 10:19:46
6750
发布2020-10-29 10:19:46
举报
文章被收录于专栏:小鹏的专栏小鹏的专栏

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

代码语言:javascript
复制
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
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2020-06-13 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档