Research on floating object classification algorithm based on convolutional neural network
Abstract With the advancement of artificial intelligence technology, unmanned boats utilizing deep learning models have shown significant potential in water surface garbage classification. This study employs Convolutional Neural Network (CNN) to extract features of water surface floating objects and...
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Nature Portfolio
2024-12-01
|
| Series: | Scientific Reports |
| Subjects: | |
| Online Access: | https://doi.org/10.1038/s41598-024-83543-9 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1841559413559656448 |
|---|---|
| author | Jikai Yang Zihan Li Ziyan Gu Wei Li |
| author_facet | Jikai Yang Zihan Li Ziyan Gu Wei Li |
| author_sort | Jikai Yang |
| collection | DOAJ |
| description | Abstract With the advancement of artificial intelligence technology, unmanned boats utilizing deep learning models have shown significant potential in water surface garbage classification. This study employs Convolutional Neural Network (CNN) to extract features of water surface floating objects and constructs the VGG16-15 model based on the VGG-16 architecture, capable of identifying 15 common types of water surface floatables. A garbage classification dataset was curated to obtain 5707 images belonging to 15 categories, which were then split into training and validation sets in a 4:1 ratio. Customized improvements were made on the base VGG-16 model, including adjusting the neural network structure to suit 15 floating object categories, applying learning rate decay and early stopping strategies for model optimization, and using data augmentation to enhance model generalization. By tweaking certain parameters, the study analyzed the impact of the number of epochs and batch sizes on the model’s classification effectiveness. The results show that the model achieves the best performance with 20 epochs and a batch size of 64, reaching a recognition accuracy of 93.86%. This is a 10.09% improvement over the traditional VGG-16 model and a 4.91% increase compared to the model without data augmentation, demonstrating the effectiveness of model improvements and data augmentation in enhancing image recognition capabilities. Additionally, the few-shot test demonstrates the fine-tuned model’s improved generalization capability. This research illustrates the applicability of transfer learning in the task of water surface garbage classification and provides technical support for the application of unmanned boats in environmental protection. |
| format | Article |
| id | doaj-art-18fa27248afe4b7c8b1ade5a6b20d552 |
| institution | Kabale University |
| issn | 2045-2322 |
| language | English |
| publishDate | 2024-12-01 |
| publisher | Nature Portfolio |
| record_format | Article |
| series | Scientific Reports |
| spelling | doaj-art-18fa27248afe4b7c8b1ade5a6b20d5522025-01-05T12:30:10ZengNature PortfolioScientific Reports2045-23222024-12-0114111410.1038/s41598-024-83543-9Research on floating object classification algorithm based on convolutional neural networkJikai Yang0Zihan Li1Ziyan Gu2Wei Li3School of Naval Architecture and Ocean Engineering, Huazhong University of Science and TechnologySchool of Naval Architecture and Ocean Engineering, Huazhong University of Science and TechnologySchool of Naval Architecture and Ocean Engineering, Huazhong University of Science and TechnologySchool of Naval Architecture and Ocean Engineering, Huazhong University of Science and TechnologyAbstract With the advancement of artificial intelligence technology, unmanned boats utilizing deep learning models have shown significant potential in water surface garbage classification. This study employs Convolutional Neural Network (CNN) to extract features of water surface floating objects and constructs the VGG16-15 model based on the VGG-16 architecture, capable of identifying 15 common types of water surface floatables. A garbage classification dataset was curated to obtain 5707 images belonging to 15 categories, which were then split into training and validation sets in a 4:1 ratio. Customized improvements were made on the base VGG-16 model, including adjusting the neural network structure to suit 15 floating object categories, applying learning rate decay and early stopping strategies for model optimization, and using data augmentation to enhance model generalization. By tweaking certain parameters, the study analyzed the impact of the number of epochs and batch sizes on the model’s classification effectiveness. The results show that the model achieves the best performance with 20 epochs and a batch size of 64, reaching a recognition accuracy of 93.86%. This is a 10.09% improvement over the traditional VGG-16 model and a 4.91% increase compared to the model without data augmentation, demonstrating the effectiveness of model improvements and data augmentation in enhancing image recognition capabilities. Additionally, the few-shot test demonstrates the fine-tuned model’s improved generalization capability. This research illustrates the applicability of transfer learning in the task of water surface garbage classification and provides technical support for the application of unmanned boats in environmental protection.https://doi.org/10.1038/s41598-024-83543-9Convolutional Neural NetworkUnmanned BoatSurface Floating DebrisImage RecognitionData Augmentation |
| spellingShingle | Jikai Yang Zihan Li Ziyan Gu Wei Li Research on floating object classification algorithm based on convolutional neural network Scientific Reports Convolutional Neural Network Unmanned Boat Surface Floating Debris Image Recognition Data Augmentation |
| title | Research on floating object classification algorithm based on convolutional neural network |
| title_full | Research on floating object classification algorithm based on convolutional neural network |
| title_fullStr | Research on floating object classification algorithm based on convolutional neural network |
| title_full_unstemmed | Research on floating object classification algorithm based on convolutional neural network |
| title_short | Research on floating object classification algorithm based on convolutional neural network |
| title_sort | research on floating object classification algorithm based on convolutional neural network |
| topic | Convolutional Neural Network Unmanned Boat Surface Floating Debris Image Recognition Data Augmentation |
| url | https://doi.org/10.1038/s41598-024-83543-9 |
| work_keys_str_mv | AT jikaiyang researchonfloatingobjectclassificationalgorithmbasedonconvolutionalneuralnetwork AT zihanli researchonfloatingobjectclassificationalgorithmbasedonconvolutionalneuralnetwork AT ziyangu researchonfloatingobjectclassificationalgorithmbasedonconvolutionalneuralnetwork AT weili researchonfloatingobjectclassificationalgorithmbasedonconvolutionalneuralnetwork |