[发明专利]一种牲畜转栏盘点方法、装置及存储介质

说明书

本发明公开了一种牲畜转栏盘点方法、装置及存储介质，其方法包括：采集目标区域中牲畜的待盘点视频数据，并将所述待盘点视频数据分为训练集和测试集；对所述测试集中的待盘点视频数据进行数据标注，得到待处理数据；采用检测算法对所述待处理数据进行训练，得到用于计算牲畜转栏数量的神经网络模型，并根据所述神经网络模型和所述测试集计算得到当前帧图像的牲畜位置；采用跟踪算法获取待盘点牲畜的运动轨迹，根据所述运动轨迹与所述目标区域的转栏线判断对所述目标区域中所有牲畜进行转栏计数。本发明实施例通过采集目标区域中牲畜的待盘点视频数据对牲畜进行转栏盘点，无需人工驱赶牲畜以及人工盘点，不仅有效提高盘点工作的工作量以及专栏盘点的准确性，还能为牲畜提供良好的生活环境。

技术领域

本发明涉及牲畜管理技术领域，尤其是涉及一种牲畜转栏盘点方法、装置及存储介质。

背景技术

在牲畜养殖行业中，随着牲畜的生长，通常需要对生长到一定重量的牲畜进行转栏处理。例如在生猪饲养环节，随着生猪的生长，原来的猪栏也已经不能够满足生猪养殖的要求，如果不区分大猪小猪，则大猪容易去伤害到小猪，小猪也容易被大猪挤压。现有的牲畜转栏盘点方法通常都是主要通过人工干预进行，通过人工驱赶牲畜到新的养殖栏并进行点数。但是这种方法存在以下问题：

(1)牲畜不配合工作人员，需要工作人员花费长时间驱赶牲畜；

(2)由人工对转栏的牲畜进行盘点，盘点工作量大且容易出现误差。

发明内容

本发明提供了一种牲畜转栏盘点方法、装置及存储介质，以解决现有的牲畜转栏盘点方法需要较大的工作量以及容易出现误差，导致盘点效率低以及误差大的技术问题。

本发明的一个实施例提供了一种牲畜转栏盘点方法，包括：

采集目标区域中牲畜的待盘点视频数据，并将所述待盘点视频数据分为训练集和测试集；

对所述测试集中的待盘点视频数据进行数据标注，得到待处理数据；

采用检测算法对所述待处理数据进行训练，得到用于计算牲畜转栏数量的神经网络模型，并根据所述神经网络模型和所述测试集计算得到当前帧图像的牲畜位置；

采用跟踪算法获取待盘点牲畜的运动轨迹，根据所述运动轨迹与所述目标区域的转栏线判断对所述目标区域中所有牲畜进行转栏计数。

进一步的，所述根据所述神经网络模型和所述测试集计算得到当前帧图像的牲畜位置，具体为：

对所述测试集中的待盘点视频数据进行数据增强处理后输入至所述神经网络模型中，计算得到所述待盘点视频数据中当前帧的牲畜位置。

进一步的，采用跟踪算法获取待盘点牲畜的运动轨迹，根据所述运动轨迹与所述目标区域的转栏线判断对所述目标区域中所有牲畜进行转栏计数，具体为：

采用跟踪算法根据所述牲畜位置将相邻两帧图像的检测框进行匹配，根据匹配结果获取待盘点牲畜的运动轨迹，根据所述运动轨迹的中心点与所述目标区域的转栏线判断对所述目标区域中所有牲畜进行转栏计数。

进一步的，所述根据所述运动轨迹的中心点与所述目标区域的转栏线判断对所述目标区域中所有牲畜进行转栏计数，具体为：

在所述中心点越过所述转栏线时，判断所述中心点越过所述转栏线的运动方向，若所述运动方向为预设的正方向，则计数加一，若所述运动方向为反方向，则计数减一。

进一步的，所述检测算法包括但不限于Yolov3算法。

进一步的，所述跟踪算法包括但不限于匈牙利算法。

进一步的，所述Yolov3算法采用CIOULoss作为目标检测损失回归函数。

本发明的第二实施例提供了一种牲畜转栏盘点装置，包括：