【更新】根据py-faster-rcnn的demo.py修改成c++版本

说明:由于改动较多,没有更新之前的博文内容,之前做的改动,请参考:根据py-faster-rcnn的demo.py修改成c++版本

★ 新版代码改了哪些内容

代码位置: https://github.com/galian123/cpp_faster_rcnn_detect

  • (1)修复了图片缩放处理的问题,原代码只是将大图片缩小,改动后可以把图片缩放到1000*600的范围内(宽或高不超1000)。既可以放大图片,也可以缩小图片。

  • (2)修复了只能检测汽车的问题。目前检测的是21种类别(除去背景,有20类),不过你可以在配置文件中修改类别数量(根据你生成的.caffemodel时设置的prototxt中的类别数量)。

  • (3) 增加yml配置文件,基本配置都放到了config目录中的yml文件中。可以创建不同的yml文件,以适应不同的训练结果。

  • (4)使用gflags处理参数。使用faster_rcnn_detect -helpshort可以查看参数的用法。 可以通过参数覆盖yml文件中的值:./faster_rcnn_detect -imgdir ../tested_images -yml_file ../config/faster_rcnn_end2end.yml GPUID 1 CONF_THRESH 0.6

  • (5)显示识别对象的名称和分数。参数-showlabel默认为true,如果显示label名,可以使用参数-noshowlabel

python代码和C++代码的详细对比, 请参考下面的链接:
————– 分割线 ————–
本系列文章如下:

猜你喜欢

转载自blog.csdn.net/u013553529/article/details/78992566
今日推荐