基于编解码网络的生活骨架提取方法研究

RESEARCH OF PIG SKELETON EXTRACTION METHOD BASED ON ENCODER-DECODER NETWORK

  • 摘要: 针对生活骨架提取难度大、精度低、耗时长等问题,提出一种基于编解码网络的生活骨架提取方法。该文构建关键点热力图生成模型,将ResNet50残差网络和U-Net语义分割网络相结合,搭建编码-解码网络结构并引入注意力机制,以提高尾、蹄等小目标关键点的特征提取精度;在生成关键点热力图的同时预测关键点偏移量,弥补反算法链点原始位置时的精度损失,利用霍夫投票机制对二者进行加权聚合,最终映射得到生活骨架。实验结果表明,骨架提取准确率为85.27%。相较于ResNet50残差网络,在耗时相近的情况下,准确率提高了22.67个百分点。该研究为生活骨架提取提供了一种新的方法,可为进一步开展生活行为研究提供技术参考。

     

    Abstract: Aimed at the problems of pig skeleton extraction, such as difficulty, low accuracy and long-time consumption, a pig skeleton extraction method based on encoder-decoder network is proposed. The key point heat map generation model was constructed, ResNet50 residual network and U-Net semantic segmentation network were combined to build an encoder-decoder network structure, and the attention mechanism was introduced to improve the feature extraction accuracy of the key points of small targets such as tail and hoof. The offset of key points was predicted while generating the key point heat map, which made up for the accuracy loss when calculating the original position of the key points. The Hough voting mechanism was used to weighted aggregate the two points, and the pig skeleton was finally mapped. The experimental results show that the skeleton extraction accuracy is 85.27%. Compared with the ResNet50 residual network, the accuracy is increased by 22.67 percentage points with similar time consumption. This study provides a new method for pig skeleton extraction, which can provide a technical reference for further pig behavior research.

     

/

返回文章
返回