We use cookies to improve your experience with our site.

基于卷积神经网络的实时多阶段斑马鱼头部姿态估计框架

A Real-Time Multi-Stage Architecture for Pose Estimation of Zebrafish Head with Convolutional Neural Networks

  • 摘要: 为了对自由游动下的斑马鱼进行光遗传学实验,必须对斑马鱼头部进行量化,以确定准确的打光位置。为了在有限的资源的设备CPU上,有效地量化斑马鱼头部的行为,我们提出了一种基于卷积神经网络的实时多阶段框架来对斑马鱼头部姿态进行估计。每个阶段都用一个小的神经网络来实现。具体来说,第一阶段使用名为Micro-YOLO的轻型目标探测器用于检测斑马鱼头部的大致区域。在第二阶段,我们设计了一个微小的包围盒优化网络,产生一个更高质量的斑马鱼头部区域。最后,设计了一个小的姿态估计网络tiny-hourglass来检测斑马鱼头部的关键点。实验结果表明,利用Micro-yolo结合RegressNet对斑马鱼头部区域进行预测,不仅比二阶段检测器Faster R-CNN更准确,而且速度更快。我们的整体框架在斑马鱼头部姿态估计方面比当前最好的用于对用户自定义区域做姿态估计的方法DeepLabCut更精准, CPU运行速度比其快19倍。

     

    Abstract: In order to conduct optical neurophysiology experiments on a freely swimming zebrafish, it is essential to quantify the zebrafish head to determine exact lighting positions. To efficiently quantify a zebrafish head's behaviors with limited resources, we propose a real-time multi-stage architecture based on convolutional neural networks for pose estimation of the zebrafish head on CPUs. Each stage is implemented with a small neural network. Specifically, a light-weight object detector named Micro-YOLO is used to detect a coarse region of the zebrafish head in the first stage. In the second stage, a tiny bounding box refinement network is devised to produce a high-quality bounding box around the zebrafish head. Finally, a small pose estimation network named tiny-hourglass is designed to detect keypoints in the zebrafish head. The experimental results show that using Micro-YOLO combined with RegressNet to predict the zebrafish head region is not only more accurate but also much faster than Faster R-CNN which is the representative of two-stage detectors. Compared with DeepLabCut, a state-of-the-art method to estimate poses for user-defined body parts, our multi-stage architecture can achieve a higher accuracy, and runs 19x faster than it on CPUs.

     

/

返回文章
返回