一种基于编解码网络的生猪骨架提取方法及评价方法
摘要:
本发明公开了一种基于编解码网络的生猪骨架提取方法及评价方法,包括以下步骤,基于ResNet50残差网络和U‑Net语义分割网络的相结合搭建编解码网络结构,并同时引入注意力机制的门控模块增强特征提取结果,生成包含位置信息的关键点热力图H;生成所述关键点热力图H的同时预测关键点偏移量,并利用霍夫投票机制将关键点偏移特征矩阵和输出的所述关键点热力图H聚合,得到精确定位的关键点位置后重新映射到原图,把关键点相连后提取出生猪骨架。本发明的有益效果:能够在光照条件不稳定、生猪运动剧烈的情况下实时的提取出生猪骨架,对于输入分辨率大小为1280x720的视频,平均均方根误差为23.43像素,骨架提取准确率为85.27%。
0/0