COCO Dataset in PyTorch

 

COCO Dataset

https://cocodataset.org/

 

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)