获取Faster RCNN最终候选框坐标值

  • Post author:
  • Post category:其他


经过之前的训练,小鱼对Faster RCNN网络实现检测功能已经了解了一点,接下来就是利用网络的检测结果实现多目标的跟踪.这个专题就用来记录一些实现跟踪道路上的小知识点.

今天小鱼分享的是:如何利用训练好的网络得到测试图片的候选框坐标?

在运行

~/py-faster-rcnn/tools/demo.py

这个文件时对测试集的目标进行了候选框的可视化,也就是

demo.py

中的

def vis_detections

函数.这里可以参考

demo.py代码解析

了解该代码的主要三个功能:候选框结果的可视化;检测目标框的获取;参数的获取

小鱼就是从这里得到最终的候选框坐标,具体方法为:

1.分析可视化什么东西

可视化函数为:

def vis_detections(im, class_name, dets, thresh=0.5):
    """Draw detected bounding boxes."""
    inds = np.where(dets[:, -1



版权声明:本文为zllljf原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。