YOLO v8 使用cityscapes数据集进行实例分割任务

一、数据集下载

Dataset之Cityscapes：Cityscapes数据集的简介、安装、使用方法之详细攻略_cityscapes dataset-CSDN博客

二、转为COCO格式

1、下载官方脚本地址：GitHub - mcordts/cityscapesScripts: README and scripts for the Cityscapes Dataset

2、cityscapesscripts/helpers/labels.py文件，找到需要的实例分割的类别。注意 hasInstances=Ture 的类别才能够用于实例分割，也就是说只有10类。

3、了解COCO格式

详解COCO数据格式的json文件内容_coco数据集json标注-CSDN博客

4、下面以其中的8类为例：

import cv2
import numpy as np
import os, glob
from shutil import copyfile
import datetime
import json
import os
import re
import fnmatch
from PIL import Image
import numpy as np
import pycococreatortools
 
ROOT_DIR = 'E:/Datesets/cityscapes/cityscapes_val'
IMAGE_DIR = os.path.join(ROOT_DIR, "images")
ANNOTATION_DIR = os.path.join(ROOT_DIR, "gt")
ANNOTATION_SAVE_DIR = os.path.join(ROOT_DIR, "annotations")
INSTANCE_DIR = os.path.join(ROOT_DIR, "instances") 
IMAGE_SAVE_DIR = os.path.join(ROOT_DIR, "val_images")

INFO = {
    "description": "Cityscapes_Instance Dataset",
    "url": "https://github.com/waspinator/pycococreator",
    "version": "0.1.0",
    "year": "2020",
    "contributor": "Kevin_Jia",
    "date_created": "2020-1-23 19:19:19.123456"
}
 
LICENSES = [
    {
        "id": 1,
        "name": "Attribution-NonCommercial-ShareAlike License",
        "url": "http://creativecommons.org/licenses/by-nc-sa/2.0/"
    }
]

CATEGORIES = [
    {
        'id': 1,
        'name': 'person',
        'supercategory': 'cityscapes',
    },
    {
        'id': 2,
        'name': 'rider',
        'supercategory': 'cityscapes',
    },
    {
        'id': 3,
        'name': 'car',
        'supercategory': 'cityscapes',
    },
    {
        'id': 4,
        'name': 'truck',
        'supercategory': 'cityscapes',
    },
    {
        'id': 5,
        'name': 'bus',
        'supercategory': 'cityscapes',
    },
    {
        'id': 6,
        'name': 'train',
        'supercategory': 'cityscapes',
    },
    {
        'id': 7,
        'name': 'motorcycle',
        'supercategory': 'cityscapes',
    },
    {
        'id': 8,
        'name': 'b