基于TensorFlow的水族馆鱼类目标检测APP开发

Aquarium fish target detection APP development based on TensorFlow

近年来深度学习在图像识别研究中取得突破进展,带动了目标检测技术的快速发展。利用目标检测技术开发水族馆鱼类目标检测APP,可以增强游客参观体验,提升科普效果。针对水族馆拍摄的80种鱼类,首先,使用LabelImg软件进行目标标记,再利用标记的目标导出成tfrecord数据;其次,选择ssd_mobilenet_v1模型进行数据训练,通过20万次的迭代训练获取到鱼类目标检测模型;最后,利用TensorFlow多目标检测API调用模型,定义2个接口和12个类,开发出Android系统手机APP。经过80种鱼类1620张图片测试,正确率为92.59%,华为MHA-AL00手机目标检测平均时间40 ms。使用鱼类目标检测APP,能实现水族馆鱼类快速识别、多鱼类目标实时检测,可提升游客的参观体验,辅助科普量化评价。

In recent years,deep learning has made a breakthrough in image recognition research,which has led to the rapid development of target detection technology.It can enhance visitors’experience and improve the effect of science popularization by using target detection technology to develop aquarium fish target detection APP.In this paper,for 80 species of fish photographed at the aquarium,firstly,LabelImg software was used to mark the target,and the marked target was exported to generate tfrecord data.Secondly,ssd_mobilenet_v1 model was selected for data training,and the fish target detection model was obtained through 200000 times of iterative training.Finally,2 interfaces and 12 classes were defined with TensorFlow multi-targets detection API call model,and Android system mobile APP was developed.After testing 1620 pictures of 80 species of fish,the accuracy rate was 92.59%.The average target detection time was 40ms on Huawei MHA-AL00.Fish target detection APP can realize fast fish recognition in aquarium and real-time detection of multiple fish targets,improve visitors’experience and contribute to quantitative evaluation of science popularization.