[发明专利]基于MBN-Transformer模型的不平衡牛身数据的双边识别方法

摘要

本发明公开了一种基于MBN‑Transformer模型的不平衡牛身数据的双边识别方法。本发明通过增加平衡采样器增大牛只少类别的采样率，通过动态融合混合增强模块减少因平衡采样器可能带来的过拟合问题，同时采用选择Vision Transformer作为共享主干网络结构，增强牛身图像全局信息的关联；采用Transformer编码器设计平衡分支和常规分支模块，分别处理牛舍随机采样器的图像混合增强数据和挤奶通道平衡采样器的图像混合增强数据，并将两个分支的输出特征通过特征动态平衡因子融合，利用Transformer编码器的多头自注意力机制挖掘动态融合混合增强模块牛身数据的关联性以及全局信息。本发明解决了挤奶通道中牛只姿态单一图像识别率性能较差问题，同时对于牛只的关键特征具有更好的聚焦能力。