yolo是目前比较流行的目标检测算法,速度快结构简单。其他的目标检测算法也有RCNN, faster-RCNN, SSD等。
近几年来,目标检测算法取得了很大的突破。比较流行的算法可以分为两类,一类是基于RegionProposal的R-CNN系算法(R-CNN, Fast R-CNN, Faster R-CNN),它们是two-stage的, 需要先使用启发式方法(selective search)或者CNN网络(RPN) 产生区域目标,然后再在区域目标上做分类与回归。而另一类是Yolo,SSD这类one-stage算法,其仅仅使用一个CNN网络直接预测不同目标的类别与位置。第一类方法是准确度高一些,但是速度慢,但是第二类算法是速度快,但是准确性要低一些。
one-stage:YOLO、SSD.
two-stage:R-CNN, Fast R-CNN, Faster R-CNN.
非极大值抑制
import numpy as np
import matplotlib