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

权利要求书

1.一种牲畜转栏盘点方法，其特征在于，包括：

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

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

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

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

2.如权利要求1所述的牲畜转栏盘点方法，其特征在于，所述根据所述神经网络模型和所述测试集计算得到当前帧图像的牲畜位置，具体为：

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

3.如权利要求1所述的牲畜转栏盘点方法，其特征在于，采用跟踪算法获取待盘点牲畜的运动轨迹，根据所述运动轨迹与所述目标区域的转栏线判断对所述目标区域中所有牲畜进行转栏计数，具体为：

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

4.如权利要求1所述的牲畜转栏盘点方法，其特征在于，所述根据所述运动轨迹的中心点与所述目标区域的转栏线判断对所述目标区域中所有牲畜进行转栏计数，具体为：

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

5.如权利要求1所述的牲畜转栏盘点方法，其特征在于，所述检测算法包括但不限于Yolov3算法。

6.如权利要求1所述的牲畜转栏盘点方法，其特征在于，所述跟踪算法包括但不限于匈牙利算法。

7.如权利要求5所述的牲畜转栏盘点方法，其特征在于，所述Yolov3算法采用CIOULoss作为目标检测损失回归函数。

8.一种牲畜转栏盘点装置，其特征在于，包括：

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

数据标注模块，用于对所述待盘点视频数据进行数据标注，得到待处理数据；

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

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

9.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质包括存储的计算机程序，其中，在所述计算机程序运行时控制所述计算机可读存储介质所在设备执行如权利要求1至7任意一项所述的牲畜转栏盘点方法。