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에서 테스트됨). |
일반적인 워크플로우
- 데이터셋 루트 설정 –
CITYSCAPES_DATASET=/path/to/cityscapes를 내보냅니다(많은 스크립트가 이 환경 변수를 읽습니다). - 다운로드 –
csDownload를 실행하여 필요한 분할(train/val/test, fine/coarse 등)을 가져옵니다. - 검사 –
csViewer를 사용하여 이미지를 탐색하고 모든 주석 유형을 오버레이합니다. - 준비 –
csCreate*유틸리티를 사용하여 다각형 주석을 모델이 기대하는 PNG 라벨 맵으로 변환합니다. - 학습 / 추론 – 준비된 이미지/라벨을 분할 또는 탐지 모델에 입력합니다.
- 평가 – 추론 후, 검증 세트에서 적절한
csEval*스크립트를 호출하여 공식 Cityscapes 지표(IoU, AP, PQ 등)를 얻습니다. - 제출 – 테스트 세트의 경우, 평가 스크립트의 설명에 따라 결과를 형식화하고 Cityscapes 제출 포털을 통해 업로드합니다.
권장 사용자
- 연구소: 최첨단 거리 장면 인식 모델을 구축하는 연구원.
- 학생: 데이터셋을 PyTorch/TensorFlow 파이프라인과 호환되는 형식으로 변환하는 신뢰할 수 있는 방법이 필요한 분.
- 경쟁자: 정확한 평가 프로토콜을 따라야 하는 Cityscapes 리더보드 참가자.
라이선스 및 연락처
- 스크립트는 MIT 라이선스입니다(수정 및 재배포 자유).
- 데이터셋 자체에는 고유한 사용자 지정 라이선스가 있습니다. 데이터를 사용하기 전에 해당 약관에 동의해야 합니다.
- 관리자: Marius Cordts, Mohamed Omran –
mail@cityscapes-dataset.net으로 연락 가능.
요약하자면, Cityscapes Scripts 저장소는 가장 널리 사용되는 자율 주행 비전 벤치마크 중 하나를 엔드 투 엔드로 처리하는 과정을 간소화하는 실용적이고 문서화가 잘 된 유틸리티 패키지입니다.
관련
- 프로젝트
- 프로젝트
- 프로젝트
- 프로젝트