Global Wheat Head Dataset#

4,698

Global Wheat 2020

ultralytics

publicDetect

Global Wheat Head Detection 2020 dataset with 4,698 unique field images across 2,675 train, 747 validation, and 1,276 test images; 3,376 labeled images, 1,322 backgrounds or unlabeled test images, 145,411 wheat-head boxes, and one wheat_head class. The documented ETHZ train/validation overlap is normalized to validation only. Created by the Global Wheat Dataset consortium and described in Global Wheat Head Detection Dataset: A Large and Diverse Dataset of High-Resolution RGB-Labelled Images to Develop and Benchmark Wheat Head Detection Methods (David et al., 2020; DOI 10.34133/2020/3521852).

1 class1.5 GBAug 21, 2026

wheat_head

agriculturewheatobject-detection+1

Il Global Wheat Head Dataset (GWHD) è un dataset di object detection a classe singola per il rilevamento delle spighe di grano — le spighe portatrici di chicchi della pianta di grano — in immagini di campi all'aperto. Fornisce 3.422 immagini di training, 748 di validazione e 1.276 di test catturate in più regioni di coltivazione ed è stato creato dalla collaborazione di nove istituti di ricerca di sette paesi affinché i modelli si generalizzino su diversi ambienti. L'accurato rilevamento delle spighe di grano è alla base delle stime di densità, dimensione e potenziale di resa delle spighe nel plant phenotyping e nella gestione delle colture.

Caratteristiche principali#

• Immagini reali di campi all'aperto che catturano la variabilità naturale nell'aspetto delle spighe di grano, nell'illuminazione e nello stadio di crescita.
• Realizzato da nove istituti di ricerca di sette paesi, che copre regioni di coltivazione europee, nordamericane, asiatiche e australiane per una solida generalizzazione tra diversi ambienti.
• Annotazioni di bounding-box per una classe singola, wheat_head, pronte per le pipeline di object detection e tracking.
• Le immagini di test provengono da genotipi e regioni non visti durante il training, fornendo un autentico benchmark di generalizzazione.

Struttura del dataset#

Il Global Wheat Head Dataset è organizzato in tre subset definiti dalla configurazione GlobalWheat2020.yaml, tutti annotati con una singola classe, wheat_head:

Applicazioni#

Il Global Wheat Head Dataset è ampiamente utilizzato per addestrare e valutare modelli di deep learning per il rilevamento delle spighe di grano. Le sue immagini diverse tra regioni, genotipi e condizioni lo rendono una risorsa preziosa per il plant phenotyping e la gestione delle colture, supportando la stima della resa, il monitoraggio della salute delle colture e l'analisi fenotipica.

Per annotare immagini di campi, addestrare e gestire le versioni dei dataset nel browser, esegui il flusso di lavoro completo con Ultralytics Platform.

Dataset YAML#

Un file YAML viene utilizzato per definire la configurazione del dataset. Definisce i percorsi, le classi e altri dettagli di configurazione del dataset. Per il Global Wheat Head Dataset, il file GlobalWheat2020.yaml è mantenuto su https://github.com/ultralytics/ultralytics/blob/main/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

Utilizzo#

Per addestrare un modello YOLO26n sul Global Wheat Head Dataset per 100 epochs con una dimensione d'immagine di 640, puoi usare i seguenti snippet di codice. Il dataset (~7.0 GB) viene scaricato automaticamente al primo utilizzo. Per un elenco completo degli argomenti disponibili, fai riferimento alla pagina di Training del modello.

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)

Esempi di immagini e annotazioni#

Il Global Wheat Head Dataset contiene una serie diversificata di immagini di campi all'aperto, che catturano la variabilità naturale nell'aspetto, negli ambienti e nelle condizioni delle spighe di grano. Ecco un'immagine di esempio dal dataset, insieme alle relative annotazioni:

• Wheat Head Detection: Le spighe di grano sono annotate con bounding box per la object detection, in una varietà di condizioni di campo che riflettono la diversità e la complessità del dataset.

Citazioni e riconoscimenti#

Se utilizzi il Global Wheat Head Dataset nel tuo lavoro di ricerca o sviluppo, ti preghiamo di citare il seguente articolo:

@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}
}

Ringraziamo i ricercatori e le istituzioni che hanno contribuito alla creazione e alla manutenzione del Global Wheat Head Dataset come risorsa preziosa per la comunità di ricerca sul plant phenotyping e sulla gestione delle colture. Per maggiori informazioni sul dataset e sui suoi creatori, visita il Global Wheat Head Dataset website.

FAQ#