图像分类、目标检测和分割数据集汇总

释放双眼,带上耳机,听听看~!
汇总了常用的图像分类、目标检测和分割数据集,包括ImageNet、COCO、LVIS、Objects365、Cityscapes和ADE20K,提供官网链接和网盘链接。

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第2天,github.com/alibaba/Eas…

图像分类

数据集汇总: ImageNet1KImageNet21K、Cifar10、Cifar100、MNIST、Fashion-MNIST、Flower102、Caltech101、Caltech256

ImageNet

官网链接:image-net.org/download.ph…

网盘链接:

ImageNet1k pan.baidu.com/s/13pKw0bJb… 提取码:0zas

ImageNet1k TFrecord pan.baidu.com/s/153SY2dp0… 提取码:5zdc

ImageNet21k pan.baidu.com/s/1eJVPCfS8… 提取码:kaeg

ImageNet是市场上最大、最受欢迎的开源数据集之一。ImageNet拥有超过1400万张已手动标注的图像。数据库按WordNet层次结构予以组织,对象级标注通过边界框完成。

图像分类、目标检测和分割数据集汇总

目标检测

数据集汇总:COCO2017、VOC2007、VOC2012LVIS、Cityscapes、Object365、CrowdHuman、OpenImages、WIDER FACE、DeepFashion、Fruit Images、Oxford-IIIT Pet、Arthropod Taxonomy Orders、African Wildlife、AI-TOD航空图、TinyPerson、WiderPerson、Caltech Pedestrian Dataset、DOTA

COCO2017

官网链接:cocodataset.org/#home

网盘链接:pan.baidu.com/s/14rO11v1V… 提取码:bcmm

COCO 是一个大型图像数据集,其被用于机器视觉领域的对象检测与分割、人物关键点检测、填充分割与字幕生成。该数据集以场景理解为主,图像中的目标则通过精确的分割进行位置标定。

该数据集具有目标分割、情景感知和超像素分割三个特征,其包含 33 万张图像、150 万目标实例、80 个目标类、91 个物品类以及 25 万关键点人物。

图像分类、目标检测和分割数据集汇总

LVIS

官网链接:www.lvisdataset.org/dataset

网盘链接:pan.baidu.com/s/1UntujlgD… 提取码:8ief

大规模的词汇实例分割数据集(Large Vocabulary Instance Segmentation,LVIS ),包含了164k图像,并针对超过1000类物体进行了约200万个高质量的实例分割标注。由于数据集中包含自然图像中的物体分布天然具有长尾属性。

图像分类、目标检测和分割数据集汇总

Objects365

官网链接:www.objects365.org/overview.ht…

该数据集总共包含63万张图像,覆盖365个类别,高达1000万框数,具有规模大、质量高、泛化能力强的特点,远超Pascal VOC、COCO等传统数据集。

图像分类、目标检测和分割数据集汇总

分割

数据集汇总:VOC2007、VOC2012、Pascal Context、COCO-Stuff 10K、Cityscapes、ADE20K

Cityscapes

官网链接:www.cityscapes-dataset.com/

该数据集拍摄了国外多个城市街道场景图片,构建数据集,其分为三个部分,包括训练集,验证集和测试集,一共 19 个类别。

图像分类、目标检测和分割数据集汇总

ADE20K

官网链接:groups.csail.mit.edu/vision/data…

网盘链接:pan.baidu.com/s/1ZuAuZheH… 提取码:dqim

ADE20K涵盖了场景、对象、对象部分的各种注释,在某些情况下甚至是部分的部分。有25k张复杂日常场景的图像,其中包含自然空间环境中的各种对象。每个图像平均有19.5个实例和10.5个对象类。

图像分类、目标检测和分割数据集汇总

姿态估计

数据集汇总: COCO2017MPII、CrowdPose、OCHuman

MPII

官网链接:human-pose.mpi-inf.mpg.de/

网盘链接:pan.baidu.com/s/1uscGGPlU… 提取码:w6af

MPII人体姿态数据集是评价关节人体姿态估计的最先进的基准。该数据集包括大约25K张图片,其中包括超过40K名带有标注身体关节的人。这些图像是根据人类日常活动的既定分类系统收集的。总体而言,数据集涵盖410项人类活动,每张图像都有活动标签。每张图片都是从YouTube视频中提取的,并提供了前后未注释的帧。

图像分类、目标检测和分割数据集汇总

EasyCV数据集接口使用示例

设计思路

EasyCV抽象了data_source来封装不同格式的数据集,data_source输出图像相关的信息,然后通过指定dataset_type来创建不同任务类型的数据集对象进行训练。其中data_source类型包括ClsSourceImageList、DetSourceCoco、DetSourceVOC、PoseTopDownSourceCoco和SegSourceRaw等等,dataset_type类型包括RawDataset、ClsDataset、DetDataset和SegDataset等等。

以ImageNet数据集为例:

# 1. 配置imagenet数据集的config
dataset_type = 'ClsDataset'
data_train_list = 'data/imagenet_raw/meta/train_labeled.txt'
data_train_root = 'data/imagenet_raw/train/'
data_test_list = 'data/imagenet_raw/meta/val_labeled.txt'
data_test_root = 'data/imagenet_raw/val/'

dataset_type = 'ClsDataset'
data = dict(
    imgs_per_gpu=32,
    workers_per_gpu=4,
    train=dict(
        type=dataset_type,
        data_source=dict(
            list_file=data_train_list,
            root=data_train_root,
            type='ClsSourceImageList'),
        pipeline=train_pipeline),
    val=dict(
        type=dataset_type,
        data_source=dict(
            list_file=data_test_list,
            root=data_test_root,
            type='ClsSourceImageList'),
        pipeline=test_pipeline))

# 2. config实例化 
cfg = mmcv_config_fromfile(args.config)

# 3. 数据集实例化
distributed = torch.cuda.is_available(
) and torch.distributed.is_initialized()
default_args = dict(
    batch_size=cfg.data.imgs_per_gpu,
    workers_per_gpu=cfg.data.workers_per_gpu,
    distributed=distributed)
dataset = build_dataset(cfg.data.train, default_args)

其他和Imagenet格式相似的数据集,都可以通过替换data_train_list、data_train_root、data_test_list和data_test_root进行配置使用,详细的config配置方式可参考github.com/alibaba/Eas…

项目开源地址github.com/alibaba/Eas…

EasyCV往期分享

本网站的内容主要来自互联网上的各种资源,仅供参考和信息分享之用,不代表本网站拥有相关版权或知识产权。如您认为内容侵犯您的权益,请联系我们,我们将尽快采取行动,包括删除或更正。
AI教程

强化学习中的Policy-gradient算法详解

2023-12-14 17:37:14

AI教程

深度残差学习框架:解读《Deep Residual Learning for Image Recognition》

2023-12-14 17:50:14

个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索