COCO Dataset in PyTorch
COCO Dataset
instances_minival2014.json
https://dl.dropboxusercontent.com/s/o43o90bna78omob/instances_minival2014.json.zip?dl=0
instances_valminusminival.json
https://dl.dropboxusercontent.com/s/s3tw5zcg7395368/instances_valminusminival2014.json.zip?dl=0
혹은
http://datasets.d2.mpi-inf.mpg.de/hosang17cvpr/coco_minival2014.tar.gz
COCO API
def __init__
# coco 불러오기 및 Annotation 불러오기
self.coco = COCO( [Annotation의 경로] )
# Img ID List 불러오기 (List에 랜덤으로, [3524, 15123, 23, 4 ...])
self.img_id = list(self.coco.imgToAnns.keys())
self.img_id = list(self.coco.getImgIds())
# category List 불러오기 (List, [1, 2, 3, ...])
self.coco_ids = sorted(self.coco.getCatIds())
# Annotation ID List 불러오기 (List, [12, 1232, 43412, ...])
self.anno_ids = self.coco.getAnnIds()
# 특정 Image Id의 Annotation ID 불러오기
self.anno_ids = self.coco.getAnnIds(imgIds=img_id)
def __getitem__
각 index 에 대하여
# image ID 변수에 저장
img_id = self.img_id[index]
# image 특징 불러오기 (loadImgs)
('license', 'url', 'file_name', 'height', 'width', 'date_captured', 'id')
# 파일 이름
self.coco.loadImgs(ids=img_id)[0]('file_name')
# width
self.coco.loadImgs(ids=img_id)[0]('width')
# Annotaion 불러오기
anno_id = self.coco.getAnnIds(imgIds=img_id)
anno = self.coco.loadAnns(ids=anno_id)