This is the official implementation of the WACV 2025 paper WeedsGalore: A Multispectral and Multitemporal UAV-based Dataset for Crop and Weed Segmentation in Agricultural Maize Fields. WeedsGalore is a UAV-based multispectral dataset with dense annotations for crop and weed segmentation in maize fields. This repository contains code and download links for the dataset and pretrained models.
Follow this link to download the dataset. The dataset (weedsgalore-dataset
, 0.4GB) and full-field orthomosaics (weedsgalore-orthomosaic
, 12GB, GeoTIFF) can be downloaded separately.
weedsgalore-dataset
└── 2023-05-25
└── images
└── semantics
└── instances
└── logs
└── 2023-05-30
└── images
└── ...
└── ...
└── splits
└── train.txt
└── ...
└── LICENSE.txt
WeedsGalore dataset is distributed under the Creative Commons Attribution (CC BY) Licence. Please refer to the full licence text for details.
Make sure to have the following installed:
- Python:
>= 3.7
- PyTorch:
>= 1.13.0
- torchmetrics:
>= 0.11.0
- absl:
>= 1.3.0
- Pillow (PIL):
>= 9.0.1
Example: Create a conda environment and install dependencies:
conda create -n weedsgalore python=3.7.12
conda activate weedsgalore
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
conda install absl-py=1.3.0
conda install pillow=9.0.1
pip install torchmetrics==0.11.4
Run the evaluation script, replacing the flags with your paths and parameters:
src/evaluate.py --dataset_path <weedsgalore-dataset_directory> --split test --ckpt <ckpt_directory> --in_channels 5 --num_classes 6
You can download pretrained models for DeepLabv3+ here (for both MSI and RGB input, and uni-weed and multi-weed case).
This project is licensed under the Apache-2.0 License. See LICENSES folder for details.
Copyright 2024 Helmholtz Centre Potsdam - GFZ German Research Centre for Geosciences
Copyright 2024 Ekin Celikkan
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
If you use the dataset or code, please cite our paper: