YOLO Vision 2026:

Global Wheat Head Dataset#

Der Global Wheat Head Dataset (GWHD) ist ein Object Detection-Datensatz mit einer Klasse zur Erkennung von Weizenähren – den korntragenden Ähren der Weizenpflanze – in Außenfeldaufnahmen. Er bietet 3.422 Trainings-, 748 Validierungs- und 1.276 Testbilder, die in mehreren Anbauregionen aufgenommen wurden, und wurde durch die Zusammenarbeit von neun Forschungsinstituten aus sieben Ländern erstellt, damit Modelle auf verschiedene Umgebungen generalisieren können. Eine genaue Weizenährenerkennung bildet die Grundlage für Schätzungen von Ährendichte, -größe und -ertragspotenzial in der Plant Phenotyping und im Pflanzenmanagement.

Hauptfunktionen#

  • Echte Feldaufnahmen im Freien, die die natürliche Variabilität im Erscheinungsbild von Weizenähren, bei der Beleuchtung und im Wachstumsstadium erfassen.
  • Erstellt von neun Forschungsinstituten aus sieben Ländern, die europäische, nordamerikanische, asiatische und australische Anbauregionen abdecken, um eine starke Generalisierung über verschiedene Umgebungen hinweg zu gewährleisten.
  • Bounding-Box-Annotationen für eine einzelne Klasse, wheat_head, bereit für Object Detection- und Tracking-Pipelines.
  • Die Testbilder stammen von Genotypen und aus Regionen, die während des Trainings nicht verwendet wurden, und stellen somit einen echten Benchmark für die Generalisierung dar.

Datensatzstruktur#

Der Global Wheat Head Dataset ist in drei Teilmengen unterteilt, die durch die GlobalWheat2020.yaml-Konfiguration definiert sind und alle mit einer einzigen Klasse annotiert sind: wheat_head:

SplitBilderRegionen
Trainieren3,422Europa (Frankreich, UK, Schweiz), Nordamerika (Kanada)
Validation748Schweiz (ETH Zürich)
Test1,276Australien, Japan, China
Validierungs-Split

Der Validierungssatz (748 Bilder) ist die ethz_1-Teilmenge, die ebenfalls Teil der Trainingsdomänen ist – daher spiegeln die Validierungsmetriken die Leistung innerhalb der Domäne wider. Der separat gehaltene Testset aus Australien, Japan und China misst die Generalisierung auf Umgebungen, die während des Trainings nicht gesehen wurden.

Anwendungen#

Der Global Wheat Head Dataset wird häufig zum Trainieren und Evaluieren von Deep Learning-Modellen für die Weizenährenerkennung verwendet. Seine vielfältigen Bilder über verschiedene Regionen, Genotypen und Bedingungen hinweg machen ihn zu einer wertvollen Ressource für das Plant Phenotyping und das Pflanzenmanagement – er unterstützt Ertragsschätzungen, die Überwachung der Pflanzengesundheit und die phänotypische Analyse.

Um Feldaufnahmen zu annotieren, Datensatzversionen in deinem Browser zu trainieren und zu verwalten, führe den vollständigen Workflow mit Ultralytics Platform aus.

Datensatz-YAML#

Eine YAML-Datei wird verwendet, um die Datensatzkonfiguration zu definieren. Sie legt die Pfade, Klassen und andere Konfigurationsdetails des Datensatzes fest. Für den Global Wheat Head Dataset wird die GlobalWheat2020.yaml-Datei unter https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/GlobalWheat2020.yaml gepflegt.

ultralytics/cfg/datasets/GlobalWheat2020.yaml
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

# Global Wheat 2020 dataset https://www.global-wheat.com/ by University of Saskatchewan
# Documentation: https://docs.ultralytics.com/datasets/detect/globalwheat2020
# Example usage: yolo train data=GlobalWheat2020.yaml
# parent
# ├── ultralytics
# └── datasets
#     └── GlobalWheat2020 ← downloads here (7.0 GB)

# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: GlobalWheat2020 # dataset root dir
train: # train images (relative to 'path') 3422 images
  - images/arvalis_1
  - images/arvalis_2
  - images/arvalis_3
  - images/ethz_1
  - images/rres_1
  - images/inrae_1
  - images/usask_1
val: # val images (relative to 'path') 748 images (WARNING: train set contains ethz_1)
  - images/ethz_1
test: # test images (optional) 1276 images
  - images/utokyo_1
  - images/utokyo_2
  - images/nau_1
  - images/uq_1

# Classes
names:
  0: wheat_head

# Download script/URL (optional) ---------------------------------------------------------------------------------------
download: |
  from pathlib import Path

  from ultralytics.utils.downloads import download

  # Download
  dir = Path(yaml["path"])  # dataset root dir
  urls = [
      "https://zenodo.org/record/4298502/files/global-wheat-codalab-official.zip",
      "https://github.com/ultralytics/assets/releases/download/v0.0.0/GlobalWheat2020_labels.zip",
  ]
  download(urls, dir=dir)

  # Make Directories
  for p in "annotations", "images", "labels":
      (dir / p).mkdir(parents=True, exist_ok=True)

  # Move
  for p in (
      "arvalis_1",
      "arvalis_2",
      "arvalis_3",
      "ethz_1",
      "rres_1",
      "inrae_1",
      "usask_1",
      "utokyo_1",
      "utokyo_2",
      "nau_1",
      "uq_1",
  ):
      (dir / "global-wheat-codalab-official" / p).rename(dir / "images" / p)  # move to /images
      f = (dir / "global-wheat-codalab-official" / p).with_suffix(".json")  # json file
      if f.exists():
          f.rename((dir / "annotations" / p).with_suffix(".json"))  # move to /annotations

Verwendung#

Um ein YOLO26n-Modell auf dem Global Wheat Head Dataset für 100 Epochs mit einer Bildgröße von 640 zu trainieren, kannst du die folgenden Codeausschnitte verwenden. Der Datensatz (~7,0 GB) wird bei der ersten Verwendung automatisch heruntergeladen. Eine umfassende Liste der verfügbaren Argumente findest du auf der Modellseite Training.

Trainingsbeispiel
from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="GlobalWheat2020.yaml", epochs=100, imgsz=640)

Beispielbilder und Annotationen#

Das Global Wheat Head Dataset enthält eine vielfältige Sammlung von Feldaufnahmen im Freien, die die natürliche Variabilität im Aussehen von Weizenähren, Umgebungen und Bedingungen erfassen. Hier ist ein Beispielbild aus dem Datensatz, zusammen mit den entsprechenden Annotationen:

Global Wheat dataset sample showing wheat head detection

  • Wheat Head Detection: Weizenähren sind mit Bounding Boxes für Object Detection über eine Vielzahl von Feldbedingungen hinweg annotiert, die die Vielfalt und Komplexität des Datensatzes widerspiegeln.

Zitate und Danksagungen#

Wenn du das Global Wheat Head Dataset in deiner Forschungs- oder Entwicklungsarbeit verwendest, zitiere bitte die folgende Publikation:

Zitat
@article{david2020global,
         title={Global Wheat Head Detection (GWHD) Dataset: A Large and Diverse Dataset of High-Resolution RGB-Labelled Images to Develop and Benchmark Wheat Head Detection Methods},
         author={David, Etienne and Madec, Simon and Sadeghi-Tehran, Pouria and Aasen, Helge and Zheng, Bangyou and Liu, Shouyang and Kirchgessner, Norbert and Ishikawa, Goro and Nagasawa, Koichi and Badhon, Minhajul and others},
         journal={arXiv preprint arXiv:2005.02162},
         year={2020}
}

Wir möchten den Forschern und Institutionen danken, die zur Erstellung und Pflege des Global Wheat Head Dataset als wertvolle Ressource für die Forschungsgemeinschaft im Bereich Pflanzenphänotypisierung und Pflanzenmanagement beigetragen haben. Weitere Informationen über den Datensatz und seine Ersteller findest du auf der Global Wheat Head Dataset website.

FAQ#

  • Der Global Wheat Head Dataset wird hauptsächlich zur Entwicklung und zum Training von Deep Learning-Modellen verwendet, die auf die Weizenährenerkennung abzielen. Dies ist entscheidend für Anwendungen in der Wheat Phenotyping und im Pflanzenmanagement, da es genauere Schätzungen der Weizenährendichte, -größe und des Gesamt-Ertragspotenzials der Ernte ermöglicht. Genaue Erkennungsmethoden helfen bei der Beurteilung von Pflanzengesundheit und Reife, die für ein effizientes Pflanzenmanagement unerlässlich sind.

  • Der Global Wheat Head Dataset hat eine einzige Klasse, wheat_head, und ist in drei Teilmengen unterteilt: 3.422 Trainingsbilder, 748 Validierungsbilder und 1.276 Testbilder. Trainings- und Validierungsbilder stammen aus Europa und Nordamerika, während der Testset aus Australien, Japan und China stammt, um die Generalisierung auf unbekannte Umgebungen zu bewerten.

  • Um ein YOLO26n-Modell auf dem Global Wheat Head Dataset zu trainieren, kannst du die folgenden Codeausschnitte verwenden. Stelle sicher, dass du die Konfigurationsdatei GlobalWheat2020.yaml hast, die die Datensatzpfade und -klassen angibt:

    Trainingsbeispiel
    from ultralytics import YOLO
    
    # Load a pretrained model (recommended for training)
    model = YOLO("yolo26n.pt")
    
    # Train the model
    results = model.train(data="GlobalWheat2020.yaml", epochs=100, imgsz=640)

    Eine umfassende Liste der verfügbaren Argumente findest du auf der Modellseite Training.

  • Der Datensatz (~7,0 GB) wird beim ersten Training mit data="GlobalWheat2020.yaml" automatisch heruntergeladen – es ist kein manueller Schritt erforderlich. Ultralytics ruft die Bilder und Labels ab und entpackt sie in dein lokales Datensatzverzeichnis. Du kannst verwandte Datensätze in der Detection Datasets Overview durchsuchen.

  • Die YAML-Konfigurationsdatei für den Global Wheat Head Dataset mit dem Namen GlobalWheat2020.yaml ist auf GitHub verfügbar. Du kannst unter https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/GlobalWheat2020.yaml darauf zugreifen. Diese Datei enthält die Datensatzpfade, Klassen und andere Konfigurationsdetails, die für das Modelltraining in Ultralytics YOLO erforderlich sind.

  • Die Weizenährenerkennung ist im Pflanzenmanagement von entscheidender Bedeutung, da sie eine genaue Schätzung der Weizenährendichte und -größe ermöglicht, die für die Bewertung der Pflanzengesundheit, der Reife und des Ertragspotenzials unerlässlich sind. Durch den Einsatz von Deep-Learning-Modellen, die auf Datensätzen wie dem Global Wheat Head Dataset trainiert wurden, können Landwirte und Forscher Kulturen besser überwachen und verwalten, was zu einer verbesserten Produktivität und einer optimierten Ressourcennutzung in der Landwirtschaft führt. Dieser technologische Fortschritt unterstützt Initiativen für Sustainable Agriculture und Ernährungssicherheit.

    Weitere Informationen zu Anwendungen von KI in der Landwirtschaft findest du unter AI in Agriculture.

Kommentare