全景分割调研(2) 常用数据集

10 篇文章 2 订阅

0. 前言

  • 全景分割调研系列文章分为4篇
    • 问题引入与性能指标
    • 常用数据集(本文)
    • 当前研究现状(现有模型分类以及当前常见研究方向)
  • 根据 paperswithcode 中描述,常用的全景分割数据集包括 COCO/Cityscapes/Mapillary/KITTI/IDD,下面分别介绍

1 COCO

  • 官网Github
  • 官网直接下载(只不过如果没有梯子,随缘进官网)
  • 数据:就是COCO2017的train/val/test数据集
  • 类别:
    • 前景(things)就是COCO检测的80类
    • 背景(stuff)就是 COCO Stuff segmentation 任务中的子集,共36类,36类经过合并后共17类。
    • 具体类别查看附录。

2 Cityscapes

  • 官网Github
  • 分割领域最常用的数据集之一。
  • 官网注册(可能要用学校邮箱,企业邮箱好像有点问题)并下载。
  • 提供的标签是instance segmentation的标签,如果用于 panoptic segmentation,则需要用到 csCreatePanopticImgs 工具。
  • 数据量:共5000张图片,共30类物体,50个城市,不同季节,白天,较好的天气,手动筛选视频帧。
  • 标签详解
    • 代码参考这里
    • 前景和背景猜测根据 hasInstances 列划分。
  • 类别查看附录

3 Mapillary

4 KITTI

  • SemanticKITTI
    • 该数据集是基于 Kitti odometry 的,换句话说,原始图像要下载 odometry 的 color数据。分割标签在前面的官网中下载。
    • sequence 00-10 有详细标注结果,11-21 作为 test set,在 evalution server 上使用。
    • 这里的全景分割数据是基于点云的,不是基于RGB图像的
    • 分类类别:28类

image-20210702190825438

  • EfficientPS
    • EfficientPS 论文的作者选择了 Kitti 中的部分图片,标注对应的全景分割标签。
    • 标签文件在上面的官网中可以直接下载得到,图片应该是从Kitti中选的吧,不知道情况如何。
    • 共有1055张图片,855训练集,200验证集。
    • 图片分辨率 1280*384
    • 11类背景,8类前景,类别从 Cityscapes 中选择

5 Indian Driving Dataset

  • 简称IDD,官网

  • 基本情况

    • 是一个在印度拍摄的自动驾驶场景数据集
    • 免费注册,无需认证,免费下载数据集。
    • 数据集有很多部分,包括了分割、检测等,目前只研究了分割,其他的没细看。
  • 分割数据集基本情况

    • 摄像头是装在车上,向前拍摄。
    • 图像大部分是1080p,少部分是720p。
    • 类别分为四个层级,具体信息参考下面配图。全景分割是,前景(things)类别就是所有 vehicles 和 living things 中的所有类别,其他的都是背景(stuff)
    • 数据集图片、视频分布如下

    img

image-20210702153429639

附录

COCO 数据集类别

[
    {
        "supercategory":"person",
        "color":[
            220,
            20,
            60
        ],
        "isthing":1,
        "id":1,
        "name":"person"
    },
    {
        "supercategory":"vehicle",
        "color":[
            119,
            11,
            32
        ],
        "isthing":1,
        "id":2,
        "name":"bicycle"
    },
    {
        "supercategory":"vehicle",
        "color":[
            0,
            0,
            142
        ],
        "isthing":1,
        "id":3,
        "name":"car"
    },
    {
        "supercategory":"vehicle",
        "color":[
            0,
            0,
            230
        ],
        "isthing":1,
        "id":4,
        "name":"motorcycle"
    },
    {
        "supercategory":"vehicle",
        "color":[
            106,
            0,
            228
        ],
        "isthing":1,
        "id":5,
        "name":"airplane"
    },
    {
        "supercategory":"vehicle",
        "color":[
            0,
            60,
            100
        ],
        "isthing":1,
        "id":6,
        "name":"bus"
    },
    {
        "supercategory":"vehicle",
        "color":[
            0,
            80,
            100
        ],
        "isthing":1,
        "id":7,
        "name":"train"
    },
    {
        "supercategory":"vehicle",
        "color":[
            0,
            0,
            70
        ],
        "isthing":1,
        "id":8,
        "name":"truck"
    },
    {
        "supercategory":"vehicle",
        "color":[
            0,
            0,
            192
        ],
        "isthing":1,
        "id":9,
        "name":"boat"
    },
    {
        "supercategory":"outdoor",
        "color":[
            250,
            170,
            30
        ],
        "isthing":1,
        "id":10,
        "name":"traffic light"
    },
    {
        "supercategory":"outdoor",
        "color":[
            100,
            170,
            30
        ],
        "isthing":1,
        "id":11,
        "name":"fire hydrant"
    },
    {
        "supercategory":"outdoor",
        "color":[
            220,
            220,
            0
        ],
        "isthing":1,
        "id":13,
        "name":"stop sign"
    },
    {
        "supercategory":"outdoor",
        "color":[
            175,
            116,
            175
        ],
        "isthing":1,
        "id":14,
        "name":"parking meter"
    },
    {
        "supercategory":"outdoor",
        "color":[
            250,
            0,
            30
        ],
        "isthing":1,
        "id":15,
        "name":"bench"
    },
    {
        "supercategory":"animal",
        "color":[
            165,
            42,
            42
        ],
        "isthing":1,
        "id":16,
        "name":"bird"
    },
    {
        "supercategory":"animal",
        "color":[
            255,
            77,
            255
        ],
        "isthing":1,
        "id":17,
        "name":"cat"
    },
    {
        "supercategory":"animal",
        "color":[
            0,
            226,
            252
        ],
        "isthing":1,
        "id":18,
        "name":"dog"
    },
    {
        "supercategory":"animal",
        "color":[
            182,
            182,
            255
        ],
        "isthing":1,
        "id":19,
        "name":"horse"
    },
    {
        "supercategory":"animal",
        "color":[
            0,
            82,
            0
        ],
        "isthing":1,
        "id":20,
        "name":"sheep"
    },
    {
        "supercategory":"animal",
        "color":[
            120,
            166,
            157
        ],
        "isthing":1,
        "id":21,
        "name":"cow"
    },
    {
        "supercategory":"animal",
        "color":[
            110,
            76,
            0
        ],
        "isthing":1,
        "id":22,
        "name":"elephant"
    },
    {
        "supercategory":"animal",
        "color":[
            174,
            57,
            255
        ],
        "isthing":1,
        "id":23,
        "name":"bear"
    },
    {
        "supercategory":"animal",
        "color":[
            199,
            100,
            0
        ],
        "isthing":1,
        "id":24,
        "name":"zebra"
    },
    {
        "supercategory":"animal",
        "color":[
            72,
            0,
            118
        ],
        "isthing":1,
        "id":25,
        "name":"giraffe"
    },
    {
        "supercategory":"accessory",
        "color":[
            255,
            179,
            240
        ],
        "isthing":1,
        "id":27,
        "name":"backpack"
    },
    {
        "supercategory":"accessory",
        "color":[
            0,
            125,
            92
        ],
        "isthing":1,
        "id":28,
        "name":"umbrella"
    },
    {
        "supercategory":"accessory",
        "color":[
            209,
            0,
            151
        ],
        "isthing":1,
        "id":31,
        "name":"handbag"
    },
    {
        "supercategory":"accessory",
        "color":[
            188,
            208,
            182
        ],
        "isthing":1,
        "id":32,
        "name":"tie"
    },
    {
        "supercategory":"accessory",
        "color":[
            0,
            220,
            176
        ],
        "isthing":1,
        "id":33,
        "name":"suitcase"
    },
    {
        "supercategory":"sports",
        "color":[
            255,
            99,
            164
        ],
        "isthing":1,
        "id":34,
        "name":"frisbee"
    },
    {
        "supercategory":"sports",
        "color":[
            92,
            0,
            73
        ],
        "isthing":1,
        "id":35,
        "name":"skis"
    },
    {
        "supercategory":"sports",
        "color":[
            133,
            129,
            255
        ],
        "isthing":1,
        "id":36,
        "name":"snowboard"
    },
    {
        "supercategory":"sports",
        "color":[
            78,
            180,
            255
        ],
        "isthing":1,
        "id":37,
        "name":"sports ball"
    },
    {
        "supercategory":"sports",
        "color":[
            0,
            228,
            0
        ],
        "isthing":1,
        "id":38,
        "name":"kite"
    },
    {
        "supercategory":"sports",
        "color":[
            174,
            255,
            243
        ],
        "isthing":1,
        "id":39,
        "name":"baseball bat"
    },
    {
        "supercategory":"sports",
        "color":[
            45,
            89,
            255
        ],
        "isthing":1,
        "id":40,
        "name":"baseball glove"
    },
    {
        "supercategory":"sports",
        "color":[
            134,
            134,
            103
        ],
        "isthing":1,
        "id":41,
        "name":"skateboard"
    },
    {
        "supercategory":"sports",
        "color":[
            145,
            148,
            174
        ],
        "isthing":1,
        "id":42,
        "name":"surfboard"
    },
    {
        "supercategory":"sports",
        "color":[
            255,
            208,
            186
        ],
        "isthing":1,
        "id":43,
        "name":"tennis racket"
    },
    {
        "supercategory":"kitchen",
        "color":[
            197,
            226,
            255
        ],
        "isthing":1,
        "id":44,
        "name":"bottle"
    },
    {
        "supercategory":"kitchen",
        "color":[
            171,
            134,
            1
        ],
        "isthing":1,
        "id":46,
        "name":"wine glass"
    },
    {
        "supercategory":"kitchen",
        "color":[
            109,
            63,
            54
        ],
        "isthing":1,
        "id":47,
        "name":"cup"
    },
    {
        "supercategory":"kitchen",
        "color":[
            207,
            138,
            255
        ],
        "isthing":1,
        "id":48,
        "name":"fork"
    },
    {
        "supercategory":"kitchen",
        "color":[
            151,
            0,
            95
        ],
        "isthing":1,
        "id":49,
        "name":"knife"
    },
    {
        "supercategory":"kitchen",
        "color":[
            9,
            80,
            61
        ],
        "isthing":1,
        "id":50,
        "name":"spoon"
    },
    {
        "supercategory":"kitchen",
        "color":[
            84,
            105,
            51
        ],
        "isthing":1,
        "id":51,
        "name":"bowl"
    },
    {
        "supercategory":"food",
        "color":[
            74,
            65,
            105
        ],
        "isthing":1,
        "id":52,
        "name":"banana"
    },
    {
        "supercategory":"food",
        "color":[
            166,
            196,
            102
        ],
        "isthing":1,
        "id":53,
        "name":"apple"
    },
    {
        "supercategory":"food",
        "color":[
            208,
            195,
            210
        ],
        "isthing":1,
        "id":54,
        "name":"sandwich"
    },
    {
        "supercategory":"food",
        "color":[
            255,
            109,
            65
        ],
        "isthing":1,
        "id":55,
        "name":"orange"
    },
    {
        "supercategory":"food",
        "color":[
            0,
            143,
            149
        ],
        "isthing":1,
        "id":56,
        "name":"broccoli"
    },
    {
        "supercategory":"food",
        "color":[
            179,
            0,
            194
        ],
        "isthing":1,
        "id":57,
        "name":"carrot"
    },
    {
        "supercategory":"food",
        "color":[
            209,
            99,
            106
        ],
        "isthing":1,
        "id":58,
        "name":"hot dog"
    },
    {
        "supercategory":"food",
        "color":[
            5,
            121,
            0
        ],
        "isthing":1,
        "id":59,
        "name":"pizza"
    },
    {
        "supercategory":"food",
        "color":[
            227,
            255,
            205
        ],
        "isthing":1,
        "id":60,
        "name":"donut"
    },
    {
        "supercategory":"food",
        "color":[
            147,
            186,
            208
        ],
        "isthing":1,
        "id":61,
        "name":"cake"
    },
    {
        "supercategory":"furniture",
        "color":[
            153,
            69,
            1
        ],
        "isthing":1,
        "id":62,
        "name":"chair"
    },
    {
        "supercategory":"furniture",
        "color":[
            3,
            95,
            161
        ],
        "isthing":1,
        "id":63,
        "name":"couch"
    },
    {
        "supercategory":"furniture",
        "color":[
            163,
            255,
            0
        ],
        "isthing":1,
        "id":64,
        "name":"potted plant"
    },
    {
        "supercategory":"furniture",
        "color":[
            119,
            0,
            170
        ],
        "isthing":1,
        "id":65,
        "name":"bed"
    },
    {
        "supercategory":"furniture",
        "color":[
            0,
            182,
            199
        ],
        "isthing":1,
        "id":67,
        "name":"dining table"
    },
    {
        "supercategory":"furniture",
        "color":[
            0,
            165,
            120
        ],
        "isthing":1,
        "id":70,
        "name":"toilet"
    },
    {
        "supercategory":"electronic",
        "color":[
            183,
            130,
            88
        ],
        "isthing":1,
        "id":72,
        "name":"tv"
    },
    {
        "supercategory":"electronic",
        "color":[
            95,
            32,
            0
        ],
        "isthing":1,
        "id":73,
        "name":"laptop"
    },
    {
        "supercategory":"electronic",
        "color":[
            130,
            114,
            135
        ],
        "isthing":1,
        "id":74,
        "name":"mouse"
    },
    {
        "supercategory":"electronic",
        "color":[
            110,
            129,
            133
        ],
        "isthing":1,
        "id":75,
        "name":"remote"
    },
    {
        "supercategory":"electronic",
        "color":[
            166,
            74,
            118
        ],
        "isthing":1,
        "id":76,
        "name":"keyboard"
    },
    {
        "supercategory":"electronic",
        "color":[
            219,
            142,
            185
        ],
        "isthing":1,
        "id":77,
        "name":"cell phone"
    },
    {
        "supercategory":"appliance",
        "color":[
            79,
            210,
            114
        ],
        "isthing":1,
        "id":78,
        "name":"microwave"
    },
    {
        "supercategory":"appliance",
        "color":[
            178,
            90,
            62
        ],
        "isthing":1,
        "id":79,
        "name":"oven"
    },
    {
        "supercategory":"appliance",
        "color":[
            65,
            70,
            15
        ],
        "isthing":1,
        "id":80,
        "name":"toaster"
    },
    {
        "supercategory":"appliance",
        "color":[
            127,
            167,
            115
        ],
        "isthing":1,
        "id":81,
        "name":"sink"
    },
    {
        "supercategory":"appliance",
        "color":[
            59,
            105,
            106
        ],
        "isthing":1,
        "id":82,
        "name":"refrigerator"
    },
    {
        "supercategory":"indoor",
        "color":[
            142,
            108,
            45
        ],
        "isthing":1,
        "id":84,
        "name":"book"
    },
    {
        "supercategory":"indoor",
        "color":[
            196,
            172,
            0
        ],
        "isthing":1,
        "id":85,
        "name":"clock"
    },
    {
        "supercategory":"indoor",
        "color":[
            95,
            54,
            80
        ],
        "isthing":1,
        "id":86,
        "name":"vase"
    },
    {
        "supercategory":"indoor",
        "color":[
            128,
            76,
            255
        ],
        "isthing":1,
        "id":87,
        "name":"scissors"
    },
    {
        "supercategory":"indoor",
        "color":[
            201,
            57,
            1
        ],
        "isthing":1,
        "id":88,
        "name":"teddy bear"
    },
    {
        "supercategory":"indoor",
        "color":[
            246,
            0,
            122
        ],
        "isthing":1,
        "id":89,
        "name":"hair drier"
    },
    {
        "supercategory":"indoor",
        "color":[
            191,
            162,
            208
        ],
        "isthing":1,
        "id":90,
        "name":"toothbrush"
    },
    {
        "supercategory":"textile",
        "color":[
            255,
            255,
            128
        ],
        "isthing":0,
        "id":92,
        "name":"banner"
    },
    {
        "supercategory":"textile",
        "color":[
            147,
            211,
            203
        ],
        "isthing":0,
        "id":93,
        "name":"blanket"
    },
    {
        "supercategory":"building",
        "color":[
            150,
            100,
            100
        ],
        "isthing":0,
        "id":95,
        "name":"bridge"
    },
    {
        "supercategory":"raw-material",
        "color":[
            168,
            171,
            172
        ],
        "isthing":0,
        "id":100,
        "name":"cardboard"
    },
    {
        "supercategory":"furniture-stuff",
        "color":[
            146,
            112,
            198
        ],
        "isthing":0,
        "id":107,
        "name":"counter"
    },
    {
        "supercategory":"textile",
        "color":[
            210,
            170,
            100
        ],
        "isthing":0,
        "id":109,
        "name":"curtain"
    },
    {
        "supercategory":"furniture-stuff",
        "color":[
            92,
            136,
            89
        ],
        "isthing":0,
        "id":112,
        "name":"door-stuff"
    },
    {
        "supercategory":"floor",
        "color":[
            218,
            88,
            184
        ],
        "isthing":0,
        "id":118,
        "name":"floor-wood"
    },
    {
        "supercategory":"plant",
        "color":[
            241,
            129,
            0
        ],
        "isthing":0,
        "id":119,
        "name":"flower"
    },
    {
        "supercategory":"food-stuff",
        "color":[
            217,
            17,
            255
        ],
        "isthing":0,
        "id":122,
        "name":"fruit"
    },
    {
        "supercategory":"ground",
        "color":[
            124,
            74,
            181
        ],
        "isthing":0,
        "id":125,
        "name":"gravel"
    },
    {
        "supercategory":"building",
        "color":[
            70,
            70,
            70
        ],
        "isthing":0,
        "id":128,
        "name":"house"
    },
    {
        "supercategory":"furniture-stuff",
        "color":[
            255,
            228,
            255
        ],
        "isthing":0,
        "id":130,
        "name":"light"
    },
    {
        "supercategory":"furniture-stuff",
        "color":[
            154,
            208,
            0
        ],
        "isthing":0,
        "id":133,
        "name":"mirror-stuff"
    },
    {
        "supercategory":"structural",
        "color":[
            193,
            0,
            92
        ],
        "isthing":0,
        "id":138,
        "name":"net"
    },
    {
        "supercategory":"textile",
        "color":[
            76,
            91,
            113
        ],
        "isthing":0,
        "id":141,
        "name":"pillow"
    },
    {
        "supercategory":"ground",
        "color":[
            255,
            180,
            195
        ],
        "isthing":0,
        "id":144,
        "name":"platform"
    },
    {
        "supercategory":"ground",
        "color":[
            106,
            154,
            176
        ],
        "isthing":0,
        "id":145,
        "name":"playingfield"
    },
    {
        "supercategory":"ground",
        "color":[
            230,
            150,
            140
        ],
        "isthing":0,
        "id":147,
        "name":"railroad"
    },
    {
        "supercategory":"water",
        "color":[
            60,
            143,
            255
        ],
        "isthing":0,
        "id":148,
        "name":"river"
    },
    {
        "supercategory":"ground",
        "color":[
            128,
            64,
            128
        ],
        "isthing":0,
        "id":149,
        "name":"road"
    },
    {
        "supercategory":"building",
        "color":[
            92,
            82,
            55
        ],
        "isthing":0,
        "id":151,
        "name":"roof"
    },
    {
        "supercategory":"ground",
        "color":[
            254,
            212,
            124
        ],
        "isthing":0,
        "id":154,
        "name":"sand"
    },
    {
        "supercategory":"water",
        "color":[
            73,
            77,
            174
        ],
        "isthing":0,
        "id":155,
        "name":"sea"
    },
    {
        "supercategory":"furniture-stuff",
        "color":[
            255,
            160,
            98
        ],
        "isthing":0,
        "id":156,
        "name":"shelf"
    },
    {
        "supercategory":"ground",
        "color":[
            255,
            255,
            255
        ],
        "isthing":0,
        "id":159,
        "name":"snow"
    },
    {
        "supercategory":"furniture-stuff",
        "color":[
            104,
            84,
            109
        ],
        "isthing":0,
        "id":161,
        "name":"stairs"
    },
    {
        "supercategory":"building",
        "color":[
            169,
            164,
            131
        ],
        "isthing":0,
        "id":166,
        "name":"tent"
    },
    {
        "supercategory":"textile",
        "color":[
            225,
            199,
            255
        ],
        "isthing":0,
        "id":168,
        "name":"towel"
    },
    {
        "supercategory":"wall",
        "color":[
            137,
            54,
            74
        ],
        "isthing":0,
        "id":171,
        "name":"wall-brick"
    },
    {
        "supercategory":"wall",
        "color":[
            135,
            158,
            223
        ],
        "isthing":0,
        "id":175,
        "name":"wall-stone"
    },
    {
        "supercategory":"wall",
        "color":[
            7,
            246,
            231
        ],
        "isthing":0,
        "id":176,
        "name":"wall-tile"
    },
    {
        "supercategory":"wall",
        "color":[
            107,
            255,
            200
        ],
        "isthing":0,
        "id":177,
        "name":"wall-wood"
    },
    {
        "supercategory":"water",
        "color":[
            58,
            41,
            149
        ],
        "isthing":0,
        "id":178,
        "name":"water-other"
    },
    {
        "supercategory":"window",
        "color":[
            183,
            121,
            142
        ],
        "isthing":0,
        "id":180,
        "name":"window-blind"
    },
    {
        "supercategory":"window",
        "color":[
            255,
            73,
            97
        ],
        "isthing":0,
        "id":181,
        "name":"window-other"
    },
    {
        "supercategory":"plant",
        "color":[
            107,
            142,
            35
        ],
        "isthing":0,
        "id":184,
        "name":"tree-merged"
    },
    {
        "supercategory":"structural",
        "color":[
            190,
            153,
            153
        ],
        "isthing":0,
        "id":185,
        "name":"fence-merged"
    },
    {
        "supercategory":"ceiling",
        "color":[
            146,
            139,
            141
        ],
        "isthing":0,
        "id":186,
        "name":"ceiling-merged"
    },
    {
        "supercategory":"sky",
        "color":[
            70,
            130,
            180
        ],
        "isthing":0,
        "id":187,
        "name":"sky-other-merged"
    },
    {
        "supercategory":"furniture-stuff",
        "color":[
            134,
            199,
            156
        ],
        "isthing":0,
        "id":188,
        "name":"cabinet-merged"
    },
    {
        "supercategory":"furniture-stuff",
        "color":[
            209,
            226,
            140
        ],
        "isthing":0,
        "id":189,
        "name":"table-merged"
    },
    {
        "supercategory":"floor",
        "color":[
            96,
            36,
            108
        ],
        "isthing":0,
        "id":190,
        "name":"floor-other-merged"
    },
    {
        "supercategory":"ground",
        "color":[
            96,
            96,
            96
        ],
        "isthing":0,
        "id":191,
        "name":"pavement-merged"
    },
    {
        "supercategory":"solid",
        "color":[
            64,
            170,
            64
        ],
        "isthing":0,
        "id":192,
        "name":"mountain-merged"
    },
    {
        "supercategory":"plant",
        "color":[
            152,
            251,
            152
        ],
        "isthing":0,
        "id":193,
        "name":"grass-merged"
    },
    {
        "supercategory":"ground",
        "color":[
            208,
            229,
            228
        ],
        "isthing":0,
        "id":194,
        "name":"dirt-merged"
    },
    {
        "supercategory":"raw-material",
        "color":[
            206,
            186,
            171
        ],
        "isthing":0,
        "id":195,
        "name":"paper-merged"
    },
    {
        "supercategory":"food-stuff",
        "color":[
            152,
            161,
            64
        ],
        "isthing":0,
        "id":196,
        "name":"food-other-merged"
    },
    {
        "supercategory":"building",
        "color":[
            116,
            112,
            0
        ],
        "isthing":0,
        "id":197,
        "name":"building-other-merged"
    },
    {
        "supercategory":"solid",
        "color":[
            0,
            114,
            143
        ],
        "isthing":0,
        "id":198,
        "name":"rock-merged"
    },
    {
        "supercategory":"wall",
        "color":[
            102,
            102,
            156
        ],
        "isthing":0,
        "id":199,
        "name":"wall-other-merged"
    },
    {
        "supercategory":"textile",
        "color":[
            250,
            141,
            255
        ],
        "isthing":0,
        "id":200,
        "name":"rug-merged"
    }
]

Cityscapes 类别

image-20210702174529045

labels = [
    #       name                     id    trainId   category            catId     hasInstances   ignoreInEval   color
    Label(  'unlabeled'            ,  0 ,      255 , 'void'            , 0       , False        , True         , (  0,  0,  0) ),
    Label(  'ego vehicle'          ,  1 ,      255 , 'void'            , 0       , False        , True         , (  0,  0,  0) ),
    Label(  'rectification border' ,  2 ,      255 , 'void'            , 0       , False        , True         , (  0,  0,  0) ),
    Label(  'out of roi'           ,  3 ,      255 , 'void'            , 0       , False        , True         , (  0,  0,  0) ),
    Label(  'static'               ,  4 ,      255 , 'void'            , 0       , False        , True         , (  0,  0,  0) ),
    Label(  'dynamic'              ,  5 ,      255 , 'void'            , 0       , False        , True         , (111, 74,  0) ),
    Label(  'ground'               ,  6 ,      255 , 'void'            , 0       , False        , True         , ( 81,  0, 81) ),
    Label(  'road'                 ,  7 ,        0 , 'flat'            , 1       , False        , False        , (128, 64,128) ),
    Label(  'sidewalk'             ,  8 ,        1 , 'flat'            , 1       , False        , False        , (244, 35,232) ),
    Label(  'parking'              ,  9 ,      255 , 'flat'            , 1       , False        , True         , (250,170,160) ),
    Label(  'rail track'           , 10 ,      255 , 'flat'            , 1       , False        , True         , (230,150,140) ),
    Label(  'building'             , 11 ,        2 , 'construction'    , 2       , False        , False        , ( 70, 70, 70) ),
    Label(  'wall'                 , 12 ,        3 , 'construction'    , 2       , False        , False        , (102,102,156) ),
    Label(  'fence'                , 13 ,        4 , 'construction'    , 2       , False        , False        , (190,153,153) ),
    Label(  'guard rail'           , 14 ,      255 , 'construction'    , 2       , False        , True         , (180,165,180) ),
    Label(  'bridge'               , 15 ,      255 , 'construction'    , 2       , False        , True         , (150,100,100) ),
    Label(  'tunnel'               , 16 ,      255 , 'construction'    , 2       , False        , True         , (150,120, 90) ),
    Label(  'pole'                 , 17 ,        5 , 'object'          , 3       , False        , False        , (153,153,153) ),
    Label(  'polegroup'            , 18 ,      255 , 'object'          , 3       , False        , True         , (153,153,153) ),
    Label(  'traffic light'        , 19 ,        6 , 'object'          , 3       , False        , False        , (250,170, 30) ),
    Label(  'traffic sign'         , 20 ,        7 , 'object'          , 3       , False        , False        , (220,220,  0) ),
    Label(  'vegetation'           , 21 ,        8 , 'nature'          , 4       , False        , False        , (107,142, 35) ),
    Label(  'terrain'              , 22 ,        9 , 'nature'          , 4       , False        , False        , (152,251,152) ),
    Label(  'sky'                  , 23 ,       10 , 'sky'             , 5       , False        , False        , ( 70,130,180) ),
    Label(  'person'               , 24 ,       11 , 'human'           , 6       , True         , False        , (220, 20, 60) ),
    Label(  'rider'                , 25 ,       12 , 'human'           , 6       , True         , False        , (255,  0,  0) ),
    Label(  'car'                  , 26 ,       13 , 'vehicle'         , 7       , True         , False        , (  0,  0,142) ),
    Label(  'truck'                , 27 ,       14 , 'vehicle'         , 7       , True         , False        , (  0,  0, 70) ),
    Label(  'bus'                  , 28 ,       15 , 'vehicle'         , 7       , True         , False        , (  0, 60,100) ),
    Label(  'caravan'              , 29 ,      255 , 'vehicle'         , 7       , True         , True         , (  0,  0, 90) ),
    Label(  'trailer'              , 30 ,      255 , 'vehicle'         , 7       , True         , True         , (  0,  0,110) ),
    Label(  'train'                , 31 ,       16 , 'vehicle'         , 7       , True         , False        , (  0, 80,100) ),
    Label(  'motorcycle'           , 32 ,       17 , 'vehicle'         , 7       , True         , False        , (  0,  0,230) ),
    Label(  'bicycle'              , 33 ,       18 , 'vehicle'         , 7       , True         , False        , (119, 11, 32) ),
    Label(  'license plate'        , -1 ,       -1 , 'vehicle'         , 7       , False        , True         , (  0,  0,142) ),
]
Detectron2框架是一个用于计算机视觉任务的强大框架,它可以用来训练和测试多种视觉任务,包括目标检测、实例分割全景分割等。如果我们要在Detectron2框架中使用自己的COCO格式的全景分割数据集,我们可以按照以下步骤进行: 1. 数据集准备:首先,我们需要准备自己的全景分割数据集,并且将其转换为COCO格式的标注。COCO格式的标注是一种常用的标注格式,它包含了图像路径、图像尺寸、每个目标的类别、边界框和分割掩膜等信息。 2. 注册数据集:接下来,我们需要注册我们的数据集,以便Detectron2框架能够识别和加载我们的数据集。在Detectron2中,我们可以使用`register_coco_instances`函数来注册COCO格式的数据集。我们需要提供数据集的名称、数据集目录、图像和标注文件的路径,以及数据集类别和颜色映射等信息。 3. 数据集预处理:在加载数据集之前,我们可以定义一些数据预处理的操作,例如图像缩放、归一化和随机增强等,以提高训练效果。我们可以使用Detectron2提供的预处理函数或者自定义自己的预处理函数来实现。 4. 模型配置:在使用Detectron2进行训练或测试之前,我们需要配置模型的参数。我们可以指定使用的模型架构、损失函数、学习率等参数,并将其写入一个配置文件中。 5. 模型训练和测试:最后,我们可以使用Detectron2框架提供的API来进行模型的训练和测试。在训练过程中,框架会自动加载我们注册的数据集,并根据配置文件中的参数进行训练;在测试过程中,框架会加载训练好的模型,并对测试集进行预测和评估。 通过以上步骤,我们可以很方便地在Detectron2框架中使用自己的COCO格式的全景分割数据集进行训练和测试。这也是Detectron2框架的一个强大功能,它可以帮助我们快速地构建和训练自定义的计算机视觉模型。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值