YOLOv A New Breakthrough in Document Layout Analysis

Introduction

As mentioned in the previous blog post, YOLOv8 performs exceptionally well in Document Layout Analysis. I trained all models from the YOLOv8 series by DocLayNet dataset and found that even the smallest model achieves an overall mAP50-95 of 71.8, while the largest model reaches an impressive 78.7.

Recently, Ultralytics released YOLOv11, the latest iteration in their YOLO series of real-time object detectors. This new version brings significant improvements to both architecture and training methods.

? The results look promising! I decided to train all YOLOv11 models on the DocLayNet dataset again and compare them with the previous YOLOv8 series.

Training Method

For this experiment, I continued to use my repository https://github.com/ppaanngggg/yolo-doclaynet to prepare the data and train the models using my custom scripts. This approach ensures consistency in the data preparation and training process, allowing for a fair comparison between YOLOv8 and YOLOv11 models.

The training and evaluation process for YOLOv11 models is straightforward and can be executed with simple command-line instructions:

# To train the model
python train.py {base-model}

# To evaluate the model
python eval.py {path-to-your-trained-model}

Comparing the Results

Here is the detailed evaluation table comparing YOLOv8 models with YOLOv11:

I've also created a plot to illustrate the relationship between model size and score for these two series:

Conclusion

Based on the table and plot above, we can conclude

Based on the table and plot above, we can conclude that YOLOv11 models consistently outperform their YOLOv8 counterparts across all sizes. The improvements are particularly noticeable in the smaller models, with YOLOv11n achieving a 1.7% increase in mAP50-95 compared to YOLOv8n. Furthermore, YOLOv11 models generally have fewer parameters than their YOLOv8 equivalents, indicating improved efficiency in addition to better performance.

My favorite model is YOLOv11l. It's only about the same size as YOLOv8m, but it outperforms even YOLOv8x!

However, YOLOv11x shows only a slight improvement over YOLOv11l despite having twice the model size.

More

What are your thoughts on the YOLOv11 results? Have you had experience using YOLO models for document layout analysis? I'd love to hear your insights and experiences in the comments below!

References

1. YOLOv11 documentation: https://docs.ultralytics.com/models/yolo11/
2. DocLayNet GitHub repository: https://github.com/DS4SD/DocLayNet
3. My YOLO-DocLayNet GitHub project: https://github.com/ppaanngggg/yolo-doclaynet

The above is the detailed content of YOLOv A New Breakthrough in Document Layout Analysis. For more information, please follow other related articles on the PHP Chinese website!