Real-time Marine Animal Images Classification by Embedded System Based on Mobilenet and Transfer Learning

Xuefeng Liu,Zhenqing Jia,Xiaoke Hou,Min Fu,Li Ma,Qiaoqiao Sun
DOI: https://doi.org/10.1109/oceanse.2019.8867190
2019-01-01
Abstract:In marine aquaculture, the classification and identification of marine animals image is one of the important research, which can help monitor the growth and fishing of marine animals, as well as water conditions. Generally, the video and image information of marine animals transmitted by underwater cameras are identified by a human or a computer, which is hardly to do real-time processing. To classify the marine animal images efficiently and in real time, a method combining an embedded system and deep learning based on MobileNetV2 and transfer learning is proposed in this paper. Firstly, the marine animal images are colleced by an underwater robot equipped with an embedded device. Next, a MobileNetV2 model based on convolutional neural network (CNN) is constructed according to the marine animal images, which can ensure the real-time requirement. Thirdly, transfer learning is used to further improve the classification. Then, the model can be trained by the collected marine animal images. Finally, the trained model could be downloaded to the embedded device and real-time classify the marine animal images under water. To evaluate the performance of the proposed method, InceptionV3 and MobilenetV1 models are used for the comparison in the experiments, and the identification accuracy rate and average classification time are calculated. The results show that the MobilenetV2 model plus transfer learning could be a better choice for the real-time classification of the marine animal images than the other two considered models. Furthermore, the size of the MobilenetV2 model is only about 40M and suitable for the embedded device.
What problem does this paper attempt to address?