pycocotools 사용법

기린이·2021년 10월 18일
0

모아두기🔧

목록 보기
12/39
coco.loadImgs(image_id)[0] # file name, w, h

ann_ids = self.coco.getAnnIds(imgIds=image_infos['id']) # 해당 이미지의 ann ids
anns = self.coco.loadAnns(ann_ids) # anns : ann들의 imgid, segmentation, area, cat_id



coco.getAnnIds(imgIds=0) # 0번 이미지의 annotation id

len(coco.getAnnIds(catIds=1)) # 카테고리가 1인 annotation id

len(coco.getImgIds(catIds=1)) # 카테고리가 1인 img id

coco.getImgIds(imgIds=10) # index로 img id 접근 가능
coco.loadAnns([1,3]) # annotation id가 1,3의 정보 리스트로 넣을 수 있음

anns = coco.loadAnns(1)

# loadAnnas(ann_id)하면 ann 정보나옴 segmentation은 모든 픽셀의 좌표

coco.annToMask(anns[0]) # anntoMask에 ann 정보룰 넣으면 mask생성해준다.
coco.loadCats(coco.getCatIds()) # category id 리스트로 넣으면 각 카테고리 정보 

id 인자를 리스트로 줄 수 있다는 것을 주의

profile
중요한 것은 속력이 아니라 방향성, 공부하며 메모를 남기는 공간입니다.

0개의 댓글