ControlNet-involved-weed-detection

This is the official repo for the paper "Weed image augmentation by ControlNet-added stable diffusion for multiclass weed detection"

Citation

Please consider cite our work if you find this repo is helpful.

@article{
  title={Weed image augmentation by ControlNet-added stable diffusion for multiclass weed detection},
  author={Lu, Yuzhen and Deng, Boyang},
  journal={Computers and Electronics in Agriculture},
  volume={Update soon},
  pages={Update soon},
  year={2025},
  publisher={ELSEVIER}
}

Contents

1. The ControlNet-added Stable Diffusion code used for generated synthetci images. Modified from https://github.com/lllyasviel/ControlNet
2. The YOLOv8 code used for training the weed detection model. Modified from https://github.com/ultralytics/ultralytics