An improved CNN model in image classification application on water turbidity
Ying Nie, Yuqiang Chen, Jianlan Guo, Shufei Li, Yu Xiao, Wendong Gong, Ruirong Lan
Abstract
Water turbidity is an important indicator for evaluating water clarity and plays an important role in environmental protection and ecological balance. Due to the subtle changes in water turbidity images, the differences captured are often too subtle to be classified. Convolutional neural networks (CNN) are widely used in image classification and perform well in feature extraction and classification. This study explored the application of convolutional neural networks in water turbidity classification. The innovation lies in applying CNN to water turbidity images, focusing on optimizing the CNN model to improve prediction accuracy and efficiency. The study proposed four CNN models for water turbidity classification based on artificial intelligence, and adjusted the number of model layers to improve prediction accuracy. Experiments were conducted on noise-free and noisy datasets to evaluate the accuracy and running time of the models. The results show that the CNN-10 model with a dropout layer has a classification accuracy of 96.5% under noisy conditions. This study has opened up new applications of CNN in fine-grained image classification, and further demonstrated the effectiveness of convolutional neural networks in water turbidity image classification through experiments.