Home

Awesome

Deep Data Augmentation for Weed Recognition Enhancement: A Diffusion Models and Transfer Learning Based Approach

Generated Samples

Performance comparison between diffusion models and GANs.

<p align="center"> <img src="docs/figs/performance.png" alt="output_example" width="30%" height="30%"> <br> Fig.1 Performance comparison between diffusion models and state-of-the-art GANs. Best values are in bold. </p>

Diffusion Models

<p align="center"> <img src="docs/figs/samples_cottonweed.png" alt="output_example" width="80%" height="40%"> <br> Fig.2 Real (upper) and generated (lower) weed samples for CottonWeedID15 dataset. Each column represents one weed class. </p> <p align="center"> <img src="docs/figs/samples_deepweeds.png" alt="output_example" width="80%" height="40%"> <br> Fig.3 Real (upper) and generated (lower) weed samples for DeepWeeds dataset. Each column represents one weed class. </p>

GANs

BigGAN

<p align="center"> <img src="docs/figs/samples_CottonWeeds_BigGAN.png" alt="output_example" width="80%" height="40%"> <br> Fig.4 Generated weed samples for CottonWeedID15 dataset by BigGAN. Each column represents one weed class. </p> <p align="center"> <img src="docs/figs/samples_deepweeds_BigGAN.png" alt="output_example" width="80%" height="40%"> <br> Fig.5 Generated weed samples for DeepWeeds dataset by BigGAN. Each column represents one weed class. </p>

StyleGAN2

<p align="center"> <img src="docs/figs/samples_CottonWeeds_StyleGAN2.png" alt="output_example" width="80%" height="40%"> <br> Fig.6 Generated weed samples for CottonWeedID15 dataset by StyleGAN2. Each column represents one weed class. </p> <p align="center"> <img src="docs/figs/samples_deepweeds_StyleGAN2.png" alt="output_example" width="80%" height="40%"> <br> Fig.7 Generated weed samples for DeepWeeds dataset by StyleGAN2. Each column represents one weed class. </p>

StyleGAN3

<p align="center"> <img src="docs/figs/samples_CottonWeeds_StyleGAN3.png" alt="output_example" width="80%" height="40%"> <br> Fig.8 Generated weed samples for CottonWeedID15 dataset by StyleGAN3. Each column represents one weed class. </p>