行人检测和跟踪——ModelArts与Atlas 200 DK的云端协作

2020/5/3 

        最近在用Atlas 200 DK，正好在学习行人跟踪方面知识，做了个行人检测和跟踪的Demo，使用华为云ModelArts预置算法YOLOv_ResNet18做行人检测，Sort算法做跟踪，结果类似如下（因为写视频有问题，就只能逐张保存成图片了）：

        效果还行。使用的是自己做的行人数据集，大概2500张图片，近20000个行人标注，在ModelArts上用Tesla V100 32GB训练了大概4个小时，数据集比较小，轮数也不多，所以行人检测效果一般。跟踪用的是Sort算法，对于遮挡或者行人出现交集的情况，ID可能会变，后面会做优化的。考虑会用DeepSort算法，加入行人计数的功能，整理完，会开源放到GitHub和码云上的。