arXiv · 2308.14397
Ensemble of Anchor-Free Models for Robust Bangla Document Layout Segmentation
Abstract
In this research paper, we introduce a novel approach designed for the purpose of segmenting the layout of Bangla documents. Our methodology involves the utilization of a sophisticated ensemble of YOLOv8 models, which were trained for the DL Sprint 2.0 - BUET CSE Fest 2023 Competition focused on Bangla document layout segmentation. Our primary emphasis lies in enhancing various aspects of the task, including techniques such as image augmentation, model architecture, and the incorporation of model ensembles. We deliberately reduce the quality of a subset of document images to enhance the resilience of model training, thereby resulting in an improvement in our cross-validation score. By employing Bayesian optimization, we determine the optimal confidence and Intersection over Union (IoU) thresholds for our model ensemble. Through our approach, we successfully demonstrate the effectiveness of anchor-free models in achieving robust layout segmentation in Bangla documents.
Explore related subjects
Keep this discovery
U Mong Sain Chak, Md. Asib Rahman. 2023-08-28. Ensemble of Anchor-Free Models for Robust Bangla Document Layout Segmentation. https://arxiv.org/abs/2308.14397
Cite the original work for its findings. Save a collection to share your selection of sources.