mcordts/cityscapesScripts

README and scripts for the Cityscapes Dataset

Cityscapes Scripts – 빠른 개요

설명 – Cityscapes 데이터셋을 다루는 데 도움을 주는 Python 툴박스입니다. 이 데이터셋은 컴퓨터 비전 모델(의미론적 분할, 인스턴스 분할, 파노프틱 분할, 3‑D 객체 탐지 등)을 학습하고 평가하는 데 사용되는 널리 알려진 거리 장면 벤치마크입니다.

중요성 – 이 데이터셋은 규모가 크고 복잡한 폴더 레이아웃과 다양한 주석 형식(정밀/거친 다각형, PNG 라벨 맵, 3‑D 박스 등)을 가지고 있습니다. 이 스크립트들은 주석을 다운로드, 시각화, 변환 및 평가하는 번거로운 단계를 자동화하여 연구자가 모델 자체에 집중할 수 있도록 합니다.


주요 기능 (README에 설명된 대로)

영역 툴박스 제공 기능
설치 pip install cityscapesscripts (GUI 추가 기능은 cityscapesscripts[gui]를 통해 설치).
데이터 처리 csDownload – 공식 Cityscapes 패키지용 명령줄 다운로더.
시각화 csViewer – 모든 주석 유형을 오버레이할 수 있는 이미지 뷰어; csLabelTool – 데이터셋 제작자가 사용한 원본 라벨링 UI.
변환 유틸리티 • csCreateTrainIdLabelImgs – 다각형 JSON을 train‑ID 라벨이 포함된 PNG로 변환.
• csCreateTrainIdInstanceImgs – 위와 동일하지만 인스턴스 ID 포함.
• csCreatePanopticImgs – COCO 스타일의 파노프틱 형식으로 내보내기.
평가 • 픽셀 수준 의미론적 라벨링 (csEvalPixelLevelSemanticLabeling).
• 인스턴스 수준 의미론적 라벨링 (csEvalInstanceLevelSemanticLabeling).
• 파노프틱 분할 (csEvalPanopticSemanticLabeling).
• 3‑D 객체 탐지 (csEvalObjectDetection3d).
3‑D 시각화 csPlot3dDetectionResults – JSON으로 저장된 3‑D 탐지 결과를 플롯.
도우미 helpers/labels.py 및 helpers/labels_cityPersons.py는 변환/평가 도구에서 사용하는 클래스 ID, 컬러 맵 및 매핑을 정의합니다.
Cython 가속 평가 속도를 높이기 위한 선택적 Cython 빌드 (CYTHONIZE_EVAL=1 python setup.py build_ext --inplace) (Ubuntu에서 테스트됨).

일반적인 워크플로우

  1. 데이터셋 루트 설정 – CITYSCAPES_DATASET=/path/to/cityscapes를 내보냅니다(많은 스크립트가 이 환경 변수를 읽습니다).
  2. 다운로드 – csDownload를 실행하여 필요한 분할(train/val/test, fine/coarse 등)을 가져옵니다.
  3. 검사 – csViewer를 사용하여 이미지를 탐색하고 모든 주석 유형을 오버레이합니다.
  4. 준비 – csCreate* 유틸리티를 사용하여 다각형 주석을 모델이 기대하는 PNG 라벨 맵으로 변환합니다.
  5. 학습 / 추론 – 준비된 이미지/라벨을 분할 또는 탐지 모델에 입력합니다.
  6. 평가 – 추론 후, 검증 세트에서 적절한 csEval* 스크립트를 호출하여 공식 Cityscapes 지표(IoU, AP, PQ 등)를 얻습니다.
  7. 제출 – 테스트 세트의 경우, 평가 스크립트의 설명에 따라 결과를 형식화하고 Cityscapes 제출 포털을 통해 업로드합니다.

권장 사용자

  • 연구소: 최첨단 거리 장면 인식 모델을 구축하는 연구원.
  • 학생: 데이터셋을 PyTorch/TensorFlow 파이프라인과 호환되는 형식으로 변환하는 신뢰할 수 있는 방법이 필요한 분.
  • 경쟁자: 정확한 평가 프로토콜을 따라야 하는 Cityscapes 리더보드 참가자.

라이선스 및 연락처

  • 스크립트는 MIT 라이선스입니다(수정 및 재배포 자유).
  • 데이터셋 자체에는 고유한 사용자 지정 라이선스가 있습니다. 데이터를 사용하기 전에 해당 약관에 동의해야 합니다.
  • 관리자: Marius Cordts, Mohamed Omran – mail@cityscapes-dataset.net으로 연락 가능.

요약하자면, Cityscapes Scripts 저장소는 가장 널리 사용되는 자율 주행 비전 벤치마크 중 하나를 엔드 투 엔드로 처리하는 과정을 간소화하는 실용적이고 문서화가 잘 된 유틸리티 패키지입니다.

관련

  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트