PSO-ShuffleNet fish recognition method based on transfer learning#br#

#br#

Expand
  • (1 Faculty of Mechanical and Electrical Engineering, Yunnan Agriculture University, Kunming 650201, Yunnan, China; 
    2 Institute of Intelligent Manufacturing Technology, Shenzhen Polytechnic, Shenzhen 518055, Guangdong, China; 
    3 Department of Mechanical Engineering, Blekinge Institute of Technology, Karlskrona 37179, Sweden)

Online published: 2023-04-25

Abstract

 Aiming at the problem that the traditional deep learning fish recognition method has a low accuracy rate and the parameters cannot be determined adaptively during the model training process. This paper proposes an improved ShuffleNet fish identification method based on Particle Swarm Optimization (PSO) and Transfer Learning (TL). The research takes 20 species of fish as the object, uses the particle swarm algorithm to take the model's loss function as the fitness function, optimizes the two hyperparameters of batch size and learning rate, and uses the transfer learning method for training to construct TL-PSO-ShuffleNet model. The research shows that compared with the models of AlexNet, MobileNet, and ShuffleNet, the recognition accuracy rate is increased by 57.89%, 30.43%, and 23.28%, respectively. The fish identification method proposed in this paper has the characteristics of high accuracy and self-adaptive parameter setting, which provides a reference for the research on automatic fish identification.

Cite this article

ZHANG Mingchen, ZHAO Lun, SHI Jie, et al . PSO-ShuffleNet fish recognition method based on transfer learning#br#

#br#
[J]. Fishery Modernization, 2023 , 50(2) : 67 -73 . DOI: 10.3969/j.issn.1007-9580.2023.02.009

Outlines

/