基于多模态猪只估重的训练方法及装置

本申请提供一种基于多模态猪只估重的训练方法及装置，涉及图像处理领域，解决了现有技术实用性欠佳、估重准确性不足的技术问题。该方法包括：获取待估重猪只的RGB图像和深度图；将RGB图像进行灰度化、掩码分割、缩放标准化处理，得到第一通道数据；基于深度图提取猪只高度特征并进行标准化处理，得到第二通道数据；基于深度图提取猪背厚度特征并进行标准化处理，得到第三通道数据；将第一通道数据、第二通道数据和第三通道数据信息输入至估重模型，得到估计重量；其中，所述估重模型基于构建PyTorch框架后训练得到，所述PyTorch框架包括主干网络、回归头模块和损失函数模块。本申请用于猪只估重过程中。