Window10+coco数据集+PythonAPI调用

来源:互联网 发布:淘宝店怎么做链接地址 编辑:程序博客网 时间:2024/06/05 03:14

1、下载coco数据集
参考博客:http://blog.csdn.net/happyhorizion/article/details/77894205#coco数据库
2、Win10编辑 Pycocotools
参考博客:http://www.jianshu.com/p/de455d653301
3、创建demo.py源码

from pycocotools.coco import COCOimport numpy as npimport skimage.io as ioimport matplotlib.pyplot as pltimport pylabpylab.rcParams['figure.figsize'] = (8.0, 10.0)dataDir='H:/TensorFlow/coco'dataType='val2014'annFile='{}/annotations/instances_{}.json'.format(dataDir,dataType)# initialize COCO api for instance annotationscoco=COCO(annFile)# display COCO categories and supercategoriescats = coco.loadCats(coco.getCatIds())nms=[cat['name'] for cat in cats]print('COCO categories: \n{}\n'.format(' '.join(nms)))nms = set([cat['supercategory'] for cat in cats])print('COCO supercategories: \n{}'.format(' '.join(nms)))# get all images containing given categories, select one at randomcatIds = coco.getCatIds(catNms=['person','dog','skateboard']);imgIds = coco.getImgIds(catIds=catIds );imgIds = coco.getImgIds(imgIds = [324158])img = coco.loadImgs(imgIds[np.random.randint(0,len(imgIds))])[0]# load and display image# I = io.imread('%s/images/%s/%s'%(dataDir,dataType,img['file_name']))# use url to load imageI = io.imread("H:\\TensorFlow\\coco\\images\\val2014\\COCO_val2014_000000324158.jpg")plt.axis('off')plt.imshow(I)plt.show()# load and display instance annotationsplt.imshow(I); plt.axis('off')annIds = coco.getAnnIds(imgIds=img['id'], catIds=catIds, iscrowd=None)anns = coco.loadAnns(annIds)coco.showAnns(anns)# initialize COCO api for person keypoints annotationsannFile = '{}/annotations/person_keypoints_{}.json'.format(dataDir,dataType)coco_kps=COCO(annFile)# load and display keypoints annotationsplt.imshow(I); plt.axis('off')ax = plt.gca()annIds = coco_kps.getAnnIds(imgIds=img['id'], catIds=catIds, iscrowd=None)anns = coco_kps.loadAnns(annIds)coco_kps.showAnns(anns)# initialize COCO api for caption annotationsannFile = '{}/annotations/captions_{}.json'.format(dataDir,dataType)coco_caps=COCO(annFile)# load and display caption annotationsannIds = coco_caps.getAnnIds(imgIds=img['id']);anns = coco_caps.loadAnns(annIds)coco_caps.showAnns(anns)plt.imshow(I); plt.axis('off'); plt.show()

4、windows配置坑比较多。欢迎交流

原创粉丝点击