View Code? Open in Web Editor
NEW
This project forked from layout-parser/layout-model-training
The scripts for training Detectron2-based Layout Models on popular layout analysis datasets
Shell 3.30%
Python 96.70%
layout-model-training's Introduction
Scripts for training Layout Detection Models using Detectron2
- 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.
- 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
- 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.
layout-model-training's People
Contributors