인프런 커뮤니티 질문&답변

작성한 질문수

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

coco.json annotation과 학습 이미지의 filename이 불일치 처리

해결된 질문

20.10.08 05:38 작성

312

Matterport로 Multi-class Mask R-CNN을 하고 있습니다.

coco로 annotation한 json 파일의 'filename' 정보와 실제 학습 이미지가 매칭되지 않을 때,

Dataset Generation에서 처리할 수 있는 코드가 있을까요. -> 매칭되지 않는 id는 데이터셋에서 제외

학습데이터수가 15만개 이상으로 매칭되지 않는 파일이 4천개가 넘어서 일일이 수정하기에는 한계가 있네요.

답변 3

지식공유자

2020. 10. 10. 22:13

해결이 되었다니, 다행입니다.

그리고 책은 집필을 포기하였습니다. 책으로 강의 내용을 쉽게 표현하는게 너무 힘들어서요. 관심 가져주셔서 감사합니다.

질문자

2020. 10. 10. 21:21

fileList = []

for file in os.listdir(path):

fileList.append(file)

return fileList

로 실제 파일 path를 fileList에 저장 후에

class CocoLikeDataset(utils.Dataset): 에서 이미지 디렉토리 내에 파일과 일치하면 image_file_name의 Dataset 코드를 만드는 코드를 삽입했습니다.

if image_file_name in fileList:

답변 감사합니다. 근데 집필 중인 <딥러닝 컴퓨터 비전> 책은 언제 출간되나요?

지식공유자

2020. 10. 08. 13:53

안녕하십니까,

매칭되는 않는다는 의미가 filename으로 보면 해당 이미지가 디렉토리에 존재하지 않는다는 의미 인가요?

만일 그렇다면 Dataset 만들기 전에 아래와 같이 os.path.isfile()로 미리 파일이 있는지 확인한 뒤 작업을 하면 되지 않을까요? 원하시는 내용이 아니라면 다시 업데이트 부탁드립니다. 감사합니다.

import os.path

if os.path.isfile(fname):
   dataset append 작업

작성한 질문수