Mask R-CNN抢车位,快人一步!
本文来自Medium上的一篇博文:Snagging Parking Spaces with Mask R-CNN and Python,使用Python与Mask RCNN检测空闲停车位。
博主Adam Geitgey说他生活在一个大城市,常常因为找不到家附近的空闲停车位而烦恼。想想下班回家后,往往也是抢车位大战的开始,在哪个小区附近不是这样呢。。。
于是作者想,能不能使用高科技手段在自己窗户上装个摄像头,监控外面有没有空闲停车位,一旦发现,赶紧开车回来抢!
好机智啊!!
作者的窗外是这样的:
作者的思路是这样的:
主要步骤:先检测监控中的停车位,再检测有没有车,然后看看这些车有没有把车位全部占满了,如果发现有空闲车位就让系统发条短信给自己。
理想是很好,但如何知道图像中哪里是停车位呢?本想事先在画面中划好位置,不过摄像头动一下就得重来。
那就尝试识别停车位!识别停车位指示表进而找停车位?识别停车位前后的划线?
都不太行!在苦思冥想各种尝试之后,作者突然灵光一现:监控窗外的车辆,如果有车长时间停的地方就是停车位啊。
好主意呢!
这时候就需要一种靠谱的车辆检测算法了!在权衡了多种算法后,作者觉得还是Mask RCNN最靠谱,它不仅可以返回车辆的包围框,还能返回车辆的轮廓。再说有COCO数据上的车辆分割的预处理模型,可以省不少力气。
Mask R-CNN检测出车辆是这样的:
于是就有了下面这个充满高科技的抢车位系统!
代码、原文下载
看完之后你也想用这套系统抢车位吧?作者人很好,已经分享了抢车位系统的代码,可惜国内因为特殊原因连博文也看不到,不过52CV君已经把它上传到百度云了。
赞 (0)