This repository contains scripts for training a multi-label segmentation model designed to identify and segment various elements within document images, such as characters, pages, and ornaments. The implementation is based on the paper from ICPR 2021, titled "Segmentation of Generative Adversarial Network."
The GAN implementation will be added to this repository soon.
To test the segmentation model with sample images, follow these steps:
-
Ensure you have the required datasets ready.
-
Run the following command:
python segmentation.py --imgDIR ./TrainingImages/1.jpg --Train 25
--train
: Specify the number of training epochs (default value is 25).
- [Milad Omrani]