Enhancing Plant Recognition in Intelligent Agricultural Irrigation Systems Using Stable Diffusion and Lora Models with CFloss
This paper proposes a novel approach to optimize the plant recognition function of intelligent agricultural irrigation system, utilizing Stable Diffusion model and Lora (Low-Rank Adaptation) model. Stable Diffusion model is a latent diffusion model that converts text into images, while Lora is a method of fine-tuning large models by reducing trainable parameters while maintaining model performance as much as possible. By combining the Stable Diffusion model and Lora model, this paper trains custom models for each plant species using a small amount of data and generates corresponding images of each variety to assist training, thus optimizing classification results. In addition, to address the issue of imbalanced data, this study also introduces Focal Loss to mitigate the impact of both class imbalance and difficulty in classification. Building upon Focal Loss, CFloss is created as the model's loss function. This method was tested on the OxfordFlower102 dataset used by the original model, and the results demonstrate that it effectively improves classification accuracy and significantly enhances the plant recognition function of the irrigation system.
Authors
- Qin Zhou (ORCID: https://orcid.org/0000-0001-5334-7188)
- Yawen Fan (ORCID: https://orcid.org/0000-0002-8316-0864)
- Chengyu Tang
- Yingzhu Lu
Institutions
- East China University of Science and Technology (CN)
- Nanjing Forestry University (CN)
- Nanjing University of Posts and Telecommunications (CN)
Publication Details
- Published
- 2024-10-26
- DOI
- https://doi.org/10.1109/cisp-bmei64163.2024.10906160
- Primary Topic
- Smart Agriculture and AI
- Type
- article
- Field-Weighted Citation Impact
- 0.00