dataset convertor
V 0.0.2 Release
PASCAL VOC : PASCAL VOC 데이터 세트에는 각 이미지에 대한 XML 파일이 있습니다.
YOLO : Yolo DataSet에는 각 이미지에 대한 TXT 파일이 있습니다.
COCO : Coco Dataset에는 각 이미지에 대한 JSON 파일이 있습니다.
현재 다음 형식이 지원됩니다.
| ~에서 | 에게 | 구현 |
|---|---|---|
| 파스칼 voc | Yolo (TXT 파일) | 예 |
| Yolo | Pascal Voc (XML 파일) | 예 |
| ~에서 | 에게 | 문제/PR (있는 경우) |
|---|---|---|
| 파스칼 voc | 코코 (JSON 파일) | 아니요 |
| 파스칼 voc | tfrecord (tfrecord 파일) | 아니요 |
| 머리 | Pascal Voc (XML 파일) | 아니요 |
| 머리 | Yolo (TXT 파일) | 아니요 |
| 머리 | tfrecord (tfrecord 파일) | 아니요 |
| Yolo | 코코 (JSON 파일) | 아니요 |
| Yolo | tfrecord (tfrecord 파일) | 아니요 |
git clone https://github.com/codePerfectPlus/dataset-convertor/ cd dataset-convertor
python -m venv venv source venv/bin/activate
pip install requirements.txtpip install dataset-convertor주석을 한 형식에서 다른 형식으로 변환하십시오.
데이터 세트 서식 예 :
- data/pascal_voc/JPEGImages/*.jpg
- data/pascal_voc/Annotations/*.xml
- data/yolo5/JPEGImages/*.jpg
- data/yolo5/labels/*.txt
from convert import Convertor
con = Convertor(input_folder= ' /home/user/data/pascal_voc ' , output_folder= ' /home/user/data/yolo5 ' )
con.voc2yolo ()from convert import Convertor
con = Convertor(input_folder= ' /home/user/data/yolo5 ' , output_folder= ' /home/user/data/pascal_voc ' )
con.yolo2voc ()형식이 없으면 문제/PR을 만듭니다. 오렌지 소스 기부금을 환영합니다. 자세한 내용은 기여 가이드를 확인하십시오.
