本周工作:完成了对于新检测模型的代码修改。
以往的检测模式只存在有三个分类结果:
person
:人体overwater
:人头在水上underwater
:人头在水下
新模型则添加了新的人体分类。
body_overwater
:人体在水上body_underwater
:人体在水下head_overwater
:人头在水上head_underwater
:人头在水下
添加对人体的水上和水下分类是为了区分该目标是否在泳池内或在岸上。这样方便我们对于泳池内总人数的计算。
在改动代码的过程中发现,人头人体匹配算法中就需要对所有目标的类别进行区分。在区分过程中发现代码的复用性比较差,因此为了提高效率,又重新对匹配算法进行了优化。
目前线下测试代码已经修改完成,接下来就是将线上测试代码进行修改。