High-Precision Fish Pose Estimation Method Based on Improved HRNet

Fish pose estimation (FPE) provides fish physiological information, facilitating health monitoring in aquaculture. It aids decision-making in areas such as fish behavior recognition. When fish are injured or deficient, they often display abnormal behaviors and noticeable changes in the positioning of their body parts. Moreover, the unpredictable posture and orientation of fish during swimming, combined with the rapid swimming speed of fish, restrict the current scope of research in FPE. In this research, a FPE model named HPFPE is presented to capture the swimming posture of fish and accurately detect their key points.

On the one hand, this model incorporated the CBAM module into the HRNet framework. The attention module enhanced accuracy without adding computational complexity, while effectively capturing a broader range of contextual information. On the other hand, the model incorporated dilated convolution to increase the receptive field, allowing it to capture more spatial context.

Experiments showed that compared with the baseline method, the average precision (AP) of HPFPE based on different backbones and input sizes on the oplegnathus punctatus datasets had increased by 0.62, 1.35, 1.76, and 1.28 percent point, respectively, while the average recall (AR) had also increased by 0.85, 1.50, 1.40, and 1.00, respectively. Additionally, HPFPE outperformed other mainstream methods, including Deep Pose, CPM, SCNet, and Lite-HRNet. Furthermore, when compared to other methods using the ornamental fish data, HPFPE achieved the highest AP and AR values of 52.96%, and 59.50%, respectively.

The proposed HPFPE can accurately estimate fish posture and assess their swimming patterns, serving as a valuable reference for applications such as fish behavior recognition.