CityscapeDataset으로 변경 시 오류

한 가지 시도해보고 있는 부분은, coco 포맷으로 변환할 때 쓰는 함수에서def convert_nucleus_to_coco(data_root_dir, image_ids, out_file): ..categories를 기존 '0', 'nulceus'에서 cityscapedataset 카테고리에 맞게 8개로 늘려서 train.json과 val.json을 뽑아보고자 함인데요,nucleus만 인식하게끔 하는 어떠한 코드나 패키지가 깔려있어 카테고리를 따로 인식하진 않는 것 같습니다. # images 와 annotations, categories list를 최종 Dict로 변환하고 json 형식으로 출력. categories = [{'id':1, 'name':'road'},{'id':2, 'name':'vegetation'},{'id':3, 'name':'sidewalk'}, {'id':4, 'name':'car'},{'id':5, 'name':'building'},{'id':6, 'name':'person'}, {'id':7, 'name':'sky'},{'id':8, 'name':'bicycle'}] coco_format_json = dict( images = images, annotations = annotations, categories = categories )

mask_rcnn활용, 데이터 coco 포맷 변환 오류

아! 원인을 알 것 같습니다.'convert_nucleus_to_coco' 함수에서 image_path 맨 뒤에 붙은 + '.png' 부분을 제거하니 기존 파일명과 일치하여 돌아갑니다! image_path = os.path.join(image_id_dir, 'image/' + image_id)