Join the conversation

Join the community of Machine Learners and AI enthusiasts.

Sign Up
dronefreakΒ 
posted an update 1 day ago
Post
1250
πŸš€ Excited to open-source the GWHD Wheat Head Detection Model Zoo on Hugging Face.

This release includes:

- πŸ€– YOLOv8, YOLOv11, YOLOv26 and RF-DETR object detection models trained on GWHD (Global Wheat Head Dataset), spanning nano through x-large variants across both architecture families.
- 🌾 Benchmarked on GWHD's dense, single-class wheat-head detection task β€” ~45 annotated heads per image on average, captured across multiple countries, genotypes, and growth stages, a genuinely hard small/dense-object setting.
- πŸ“Š Detailed model cards with mAP/precision/recall, per-class breakdowns, PR/F1 curves and confusion matrices (YOLO), qualitative detection showcases, and full training configurations for reproducibility.

Headline numbers:
- πŸ† Best mAP@50: 74.25% (YOLOv11x), 34.92% mAP@50:95, 83.37% precision.
- ⚑ Best efficiency tradeoff: YOLOv26s hits 70.49% mAP@50 at just 22.8 GFLOPs (10.0M params) β€” within ~4 points of the top YOewer FLOPs (196.0 GFLOPs).

The goal is to make benchmarking and experimenting with agricultural computer vision easier by providing ready-to-use pretrained checkpoints, all trained and evaluated under one shared pipeline (DetectionBench: https://github.com/dronefreak/DetectionBench).
Full credit for the underlying dataset goes to Etienne David, Mario Serouart, Simon Madec, and the Global Wheat Head Detection 2020/2021) β€” this release is anunofficial, YOLO-ready reformatting of their work, not a new dataset.

If you're working on precision at detection, or just want areproducible detector benchmark, I hope these resources are useful.

πŸ“¦ Dataset:
dronefreak/GWHD

πŸ€– Model Collection: dronefreak/gwhd-wheat-head-detection-model-zoo-6a7aea28b5431918cc46cec1

Feedback, bug reports, and contributions are always welcome.

Publishing the training configurations alongside mAP@50 and mAP@50:95 is a strong reproducibility choice for dense-object detection. A useful next comparison would be per-country or genotype slices, since domain shift can matter more than the aggregate score in field deployment. Are those stratified results included in the model cards?

Β·

Hi @Yusufchy
Thank you for your query. You make a great point in the sense that per-country or genotype slicing could indeed impact the results significantly. These stratified results are not yet published in the model cards.

However, if this is something you think could be interesting, I could work upon it and release something like that in the next release. On my side as well I was wondering what could be possible next steps, so thank you for shedding some light in this regard.