🌌 OpenImages Tool

A tool to convert OpenImages dataset format to TensorFlow-friendly format;

• ➰ Convert .txt to .xml (VOC) format
• 🚀 Make directory structure suitable to be used in TensorFlow custom object training
• 👩‍💻 Use presented Code Snippets for other organizing operations
  • string_replacer
  • xml_replacer
  • file_renamer

🏗️ Required Directory Structure

Dataset
|___ validation
|    |___ <object_name>
|          |___ Label
|          |    |___ <validation_label_file_name>.txt
|          |    |___ ....
|          |___ <validation_image_file_name>.jpg
|          |___ ....
|
|___ test
|    |___ <object_name>
|          |___ Label
|          |    |___ <validation_label_file_name>.txt
|          |    |___ ....
|          |___ <validation_image_file_name>.jpg
|          |___ ....
|
|___ train
     |___ <object_name>
           |___ Label
           |    |___ <validation_label_file_name>.txt
           |    |___ ....
           |___ <validation_image_file_name>.jpg
           |___ ....

📢 validation, test, train and Label are required fix keywords

🎉 Result Directory Structure

images
|___ test
|    |___ <object_name>_test_<id>.xml
|    |___ <object_name>_test_<id>.jpg
|    |___ ....
|
|___ train
     |___ <object_name>_train_<id>.xml
     |___ <object_name>_train_<id>.jpg
     |___ ....

👩‍💻 Usage

1. Clone this repository
2. Organize your data to be like the required
3. Open src folder in CMD
4. Run:

 src> python script.py -i <INPUT_PATH> -o <OUTPUT_PATH>

👮‍♀️ This command will add validation set to training set folder, if you want to disable this behavior then run:

 src> python script.py -i <INPUT_PATH> -o <OUTPUT_PATH> -v

👀 To see running options, run:

src> python script.py -h

📖 References

• OpenImages Dataset
• OIDv4_ToolKit
• OIDv4_to_VOC

💼 Contact & Support

Find me on LinkedIn and feel free to mail me, Asmaa 🦋