COCO라이브러리

1. coco포멧의 json파일을 가져온다.

coco=COCO('train.json')

2. image load

image_id = coco.getImgIds(imgIds=1) # ImgIds=1인 이미지의 아이디를 리스트로 가져온다.
image_id = coco.getImgIds(imgIds=[1, 2]) # ImgIds가 1과 2인 이미지의 아이디를 리스트로 가져온다.

image_id = coco.getImgIds(catIds=1) # category ID가 1인 이미지의 아이디를 리스트로 가져온다.
image_id = coco.getImgIds(catIds=[1, 2]) # category ID가 1, 2인 이미지의 아이디를 리스트로 가져온다.

image_infos = coco.loadImgs(image_id) 
# 리스트로 주어진 이미지 아이디에 대한 정보를 리스트 형태로 가져온다.
# [{'license': , 'url': , 'file_name': , 'height': , 'width': , 'date_captured': , 'id': }]

3. annotation load

ann_ids = coco.getAnnIds(imgIds=image_infos['id'])
anns = coco.loadAnns(ann_ids)

위와 마찬가지로 imgid나 category가 특정