[컴퓨터비전] KITTI dataset label (.txt) 파일을 PASCAL VOC label (.xml)로 변경하는 방법

KITTI 2 PASCAL

참고 깃허브: https://github.com/umautobots/vod-converter?tab=readme-ov-file

GitHub - umautobots/vod-converter: Convert between visual object detection datasets

 

위 깃허브에서 코드 다운받은 후, 폴더(vod-converter-master)에 아래와 같이 필요한 내용 추가하기

.idea
datasets
mydata-kitti
├── training
│   ├── image_2
│   └── label_2
└── train.txt
tests
vod_converter
.gitignore
LICENSE
README.md

 

대부분은 github코드에서 받아온 자료들이고 우리가 만들 폴더는 mydata-kitti 및 하위 폴더만 만들면 된다. 우리가 추가로 다운받을 파일은 총 3종류이다.

 

1. Image_2: training data image 7481장, Download left color images of object data set (12 GB). 해당 데이터는 KITTI 홈페이지에서 다운로드 받기: https://www.cvlibs.net/datasets/kitti/eval_object.php?obj_benchmark=2d
2. label_2: training data label 7481개, Download training labels of object data set (5 MB). 마찬가지로 KITTI 홈페이지에서 다운받은 .txt파일
3. train.txt: 동일 깃허브에서 제공하는 파일

세 종류의 파일들을 다운받아 적절한 폴더(디렉토리)에 넣은 후, 가상환경(을 추천)에서 폴더 위치 확인해준 후, 아래와 같은 명령어를 입력하면 된다.

python vod_converter/main.py --from kitti --from-path datasets/mydata-kitti --to voc --to-path datasets/mydata-vocv