WeedGeneration

Title: Channel Attention GAN-based Synthetic Weed Generation for Precise Weed Identification
Kyewords: generative adversarial network; deep learning; phenotyping; precision agriculture

Dataset

Please fill this form to get download link of the datasets and pretrained weights

Train

To train the gan model by yourself, please:

1. Download the 'dataset.zip' following the above Dataset section and unzip the "datasets/" to the root of this repo.
2. Check the configuration.py
3. Train the model by running:

   python train.py
   

Test

To generate synthetic dataset on Colab: here
To generate synthetic dataset on your own pc, simply check the arguments in test.py and run:

python test.py