点击上方“小白学视觉”,选择加"星标"或“置顶”
重磅干货,第一时间送达
通过使用深度学习和opencv,我们可以从视频剪辑中提取有趣的见解。请参阅下面的B / w UCF和Duke游戏的gif示例,在这里我们可以识别所有球员+裁判,并根据球员的球衣颜色将其标记为球队。
我们使用了像Faster RCNN这样的预训练检测模型来检测玩家。你们可以轻松地从Tensorflow Object Detection API下载针对COCO数据集训练的Faster RCNN并对其进行测试。API将每帧图像作为输入,并在80种不同的类别中进行检测。请阅读下面的样品检测示例,我们压制了太大的检测范围,无法更清晰地将玩家细分。你们也可以在API中使用得分阈值来过滤掉低置信度检测。
现在来了有趣的部分。我们如何确定UCF对阵杜克大学的球员?我们可以使用OpenCV来做到这一点。OpenCV允许我们识别特定颜色的蒙版,并且可以使用它来识别白人和黑人玩家。主要步骤是:
OpenCV Bitwise_and将蒙版中不存在的任何像素涂成黑色
请浏览下面的输出以了解白色。它们被屏蔽为“粉红色”,而其他所有内容均以黑色掩盖