Scripts for training Layout Detection Models using Detectron2

Usage

• In tools/, we provide a series of handy scripts for converting data formats and training the models.
• In scripts/, it lists specific command for running the code for processing the given dataset.
• The configs/ contains the configuration for different deep learning models, and is organized by datasets.

Supported Datasets

• Prima Layout Analysis Dataset scripts/train_prima.sh
  • You will need to download the dataset from the official website and put it in the data/prima folder.
  • As the original dataset is stored in the PAGE format, the script will use tools/convert_prima_to_coco.py to convert it to COCO format.
  • The final dataset folder structure should look like:

    data/
    └── prima/
        ├── Images/
        ├── XML/
        ├── License.txt
        └── annotations*.json

Reference

• cocosplit A script that splits the coco annotations into train and test sets.
• Detectron2 Detectron2 is Facebook AI Research's next generation software system that implements state-of-the-art object detection algorithms.