dataset convertor
V 0.0.2 Release
PASCAL VOC :Pascal Vocデータセットには、各画像にXMLファイルがあります。
YOLO :Yolo Datasetには、各画像にTXTファイルがあります。
COCO :CoCoデータセットには、各画像にJSONファイルがあります。
現在、次の形式がサポートされています。
| から | に | 実装 |
|---|---|---|
| パスカル・ヴォック | Yolo(TXTファイル) | はい |
| ヨロ | Pascal Voc(XMLファイル) | はい |
| から | に | 発行/PR(ある場合) |
|---|---|---|
| パスカル・ヴォック | coco(jsonファイル) | いいえ |
| パスカル・ヴォック | tfrecord(tfrecordファイル) | いいえ |
| ココ | Pascal Voc(XMLファイル) | いいえ |
| ココ | Yolo(TXTファイル) | いいえ |
| ココ | tfrecord(tfrecordファイル) | いいえ |
| ヨロ | coco(jsonファイル) | いいえ |
| ヨロ | 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を作成するフォーマットが欠落している場合は、オープンソースの貢献を歓迎します。詳細については、寄稿ガイドを確認してください。