coco数据集大小分类_[COCO数据集]关于instances中的分割信息按部分类别进行获取及保存...

1 #-*- coding: utf-8 -*-

7 from pycocotools.coco importCOCO8 importos9 importshutil10 from tqdm importtqdm11 importmatplotlib.pyplot as plt12 importcv213 from PIL importImage, ImageDraw14 importskimage.io as io15 importjson16

17 '''

18 路径参数19 '''

20 #原coco数据集的路径

21 dataDir= './'

22 #用于保存新生成的数据的路径

23 savepath = "newdata/"

24 #只保存含有你需要的类别的图片的路径，最后没有用

25 #因为没必要，coco是按json中的信息读图，只要在json里做筛选就行了

26 img_save = savepath + 'images/'

27 #最后生产的json文件的保存路径

28 anno_save = savepath+'annotations/'

29 '''

30 数据集参数31 '''

32 #coco有80类，这里写要提取部分类的名字

33 #如我只需要car、bus、truck这三类数据

34 classes_names = ['car','bus','truck']35 #要处理的数据集，比如val2017、train2017等