1. 程式人生 > >純C++版的Faster-Rcnn(通過caffe自定義RPN層實現)

純C++版的Faster-Rcnn(通過caffe自定義RPN層實現)

這裡介紹的是通過新增自定義層(RPN層)代替python層,實現c++版的faster-rcnn,因為去掉python了,所以部署時不會因為牽扯到python庫等其它的莫名其妙的錯誤,使用起來就跟單純的caffe一樣,更簡單方便。 核心程式碼,借鑑的是這篇部落格,這裡的話,我們不扣具體的程式碼細節(比如rpn層是怎麼產出候選框啊,非極大值抑制是具體怎麼實現的等等),有興趣的可以自己查下資料,所以主要是走一個步驟,從而完成c++版faster-rcnn的配置。

        步入正題,步驟和上面那篇部落格大致一樣,但它有一些細節地方直接忽略了,程式碼也有幾處小bug,所以我把具體的流程給說下。

      (1) 新增自定義層 rpn_layer.hpp  把它放在 caffe/include/caffe/layers/  目錄下

  1. #ifndef CAFFE_RPN_LAYER_HPP_
  2. #define CAFFE_RPN_LAYER_HPP_
  3. #include <vector>
  4. #include "caffe/blob.hpp"
  5. #include "caffe/layer.hpp"
  6. #include "caffe/proto/caffe.pb.h"
  7. //#include"opencv2/opencv.hpp"
  8. #define mymax(a,b) ((a)>(b))?(a):(b)
  9. #define mymin(a,b) ((a)>(b))?(b):(a)
  10. namespace caffe {  
  11.     /** 
  12.     * @brief implement RPN layer for faster rcnn 
  13.     */
  14.     template <typename Dtype>  
  15.     class RPNLayer : public Layer<Dtype> {  
  16.     public:  
  17.         explicit RPNLayer(const LayerParameter& param)  
  18.             : Layer<Dtype>(param) {  
  19.                 m_score_.reset(new Blob<Dtype>());  
  20.                 m_box_.reset(new Blob<Dtype>());  
  21.                 local_anchors_.reset(new Blob<Dtype>());  
  22.             }  
  23.         virtualvoid LayerSetUp(const vector<Blob<Dtype>*>& bottom,  
  24.             const vector<Blob<Dtype>*>& top);  
  25.         virtualvoid Reshape(const vector<Blob<Dtype>*>& bottom,  
  26.             const vector<Blob<Dtype>*>& top){}  
  27.         virtualinlineconstchar* type() const { return"RPN"; }  
  28.         struct abox{  
  29.             Dtype batch_ind;  
  30.             Dtype x1;  
  31.             Dtype y1;  
  32.             Dtype x2;  
  33.             Dtype y2;  
  34.             Dtype score;  
  35.             bool operator <(const abox&tmp) const{  
  36.                 return score < tmp.score;  
  37.             }  
  38.         };  
  39.     protected:  
  40.         virtualvoid Forward_cpu(const vector<Blob<Dtype>*>& bottom,  
  41.             const vector<Blob<Dtype>*>& top);  
  42.         //virtual void Forward_gpu(const vector<Blob<Dtype>*>& bottom,
  43.             //const vector<Blob<Dtype>*>& top);
  44.         virtualvoid Backward_cpu(const vector<Blob<Dtype>*>& top,  
  45.             const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom){};  
  46.         int feat_stride_;  
  47.         int base_size_;  
  48.         int min_size_;  
  49.         int pre_nms_topN_;  
  50.         int post_nms_topN_;  
  51.         float nms_thresh_;  
  52.         vector<int> anchor_scales_;  
  53.         vector<float> ratios_;  
  54.         vector<vector<float> > gen_anchors_;  
  55.         int *anchors_;  
  56.         int anchors_nums_;  
  57.         int src_height_;  
  58.         int src_width_;  
  59.         float src_scale_;  
  60.         int map_width_;  
  61.         int map_height_;  
  62.         shared_ptr<Blob<Dtype> > m_score_;  
  63.         shared_ptr<Blob<Dtype> > m_box_;  
  64.         shared_ptr<Blob<Dtype> >local_anchors_;  
  65.         void generate_anchors();  
  66.         vector<vector<float> > ratio_enum(vector<float>);  
  67.         vector<float> whctrs(vector<float>);  
  68.         vector<float> mkanchor(float w,float h,float x_ctr,float y_ctr);  
  69.         vector<vector<float> > scale_enum(vector<float>);  
  70.         //cv::Mat proposal_local_anchor(int width, int height);
  71.         void proposal_local_anchor();  
  72.         void bbox_tranform_inv();  
  73.         cv::Mat bbox_tranform_inv(cv::Mat local_anchors, cv::Mat boxs_delta);  
  74.         void nms(std::vector<abox> &input_boxes, float nms_thresh);  
  75.         void filter_boxs(cv::Mat& pre_box, cv::Mat& score, vector<abox>& aboxes);  
  76.         void filter_boxs(vector<abox>& aboxes);  
  77.     };  
  78. }  // namespace caffe
  79. #endif  // CAFFE_RPN_LAYER_HPP_
然後是原始檔 rpn_layer.cpp  放在 caffe/src/caffe/layers/  目錄下
  1. #include <algorithm>
  2. #include <vector>
  3. #include "caffe/layers/rpn_layer.hpp"
  4. #include "caffe/util/math_functions.hpp"
  5. #include <opencv2/opencv.hpp>
  6. int debug = 0;  
  7. int  tmp[9][4] = {  
  8.     { -83, -39, 100, 56 },  
  9.     { -175, -87, 192, 104 },  
  10.     { -359, -183, 376, 200 },  
  11.     { -55, -55, 72, 72 },  
  12.     { -119, -119, 136, 136 },  
  13.     { -247, -247, 264, 264 },  
  14.     { -35, -79, 52, 96 },  
  15.     { -79, -167, 96, 184 },  
  16.     { -167, -343, 184, 360 }  
  17. };  
  18. namespace

    相關推薦

    C++Faster-Rcnn通過caffe定義RPN實現

    這裡介紹的是通過新增自定義層(RPN層)代替python層,實現c++版的faster-rcnn,因為去掉python了,所以部署時不會因為牽扯到python庫等其它的莫名其妙的錯誤,使用起來就跟單純的caffe一樣,更簡單方便。 核心程式碼,借鑑的是這篇部落格,這裡

    C++500VIP源碼下載的Faster R-CNN通過caffe定義RPN實現

    方便 預測 大致 ole test cto oop 可執行文件 names 這裏500VIP源碼下載 dsluntan.com 介紹的是通過添加自定義層(RPN層)代替python層,實現c++版的Faster R-CNN,因為去掉python了,所以部署時不會因為牽扯到p

    C++Faster R-CNN通過caffe定義RPN實現

             這裡介紹的是通過新增自定義層(RPN層)代替python層,實現c++版的Faster R-CNN,因為去掉python了,所以部署時不會因為牽扯到python庫等其它的莫名其妙的錯誤,使用起來就跟單純的caffe一樣,更簡單方便。 核心程式碼,借鑑的是這篇

    C++Faster R-CNNcaffe定義RPN實現 個人見解 問題分析記錄

    本文是在實現https://blog.csdn.net/zxj942405301/article/details/72775463這篇部落格裡面的程式碼時遇到的一些問題,然後理解分析記錄在此。前面的程式碼部分按照步驟進行應該不會有錯,有一個提示的地方是在caffe.proto

    windows下的c++ Faster R-CNN

         效果如下圖: 同時版本經過大幅度效能優化,在GeForce GTX TITAN X顯示卡上,對於VGG16模型的速度是16.6fps左右。 *****************************分割線*************************

    [目標檢測]windows下實現c++faster-rcnn

    本版本根據windows下matlab版改寫而來。工程可到http://download.csdn.net/detail/oyangzi12/9692597 下載。程式預設使用GPU模式,如果沒有GPU只需在程式中將caffe設定為cpu模式。使用方法:1、配置opencv,

    windows下實現c++faster-rcnn

    本文主要參考:http://blog.csdn.net/oYangZi12/article/details/53290426?locationNum=5&fps=1  1.下載微軟提供的caffe(https://github.com/Microsoft/c

    IOS xib在tableview上的簡單應用通過xib定義cell

    make aso eat 常用 last creat path ins div UITableView是一種常用的UI控件,在實際開發中,由於原生api的局限,自定義UITableViewCell十分重要,自定義cell可以通過代碼,也可以通過xib。 這篇隨筆介紹的是通過

    web專案Log4j日誌輸出路徑配置問題 問題描述:一個web專案想在一個tomcat下執行多個例項通過修改war包名稱的實現,然後每個例項都將日誌輸出到tomcat的logs目錄下例項名命名的文

    問題描述:一個web專案想在一個tomcat下執行多個例項(通過修改war包名稱的實現),然後每個例項都將日誌輸出到tomcat的logs目錄下例項名命名的資料夾下進行區分檢視每個例項日誌,要求通過儘可能少的改動配置檔案,最好修改例項名後可以不修改log4j的配置檔案。 實現分析:一般實現上面需求,需要在修

    C# Winform 窗體美化十、定義窗體

    十、自定義窗體 寫在前面 最近在做 winform 應用程式,需要自定義一種視窗的樣式,所以就隨便搞了一個簡單的視窗。 效果圖 有兩種樣式,介面如下: 無標題: 有標題: 關鍵詞 1、黑色描邊邊框 對於視窗去掉原生的邊框

    C++ 類模板小結雙向連結串列的類模板實現

    一、類模板定義 定義一個類模板:template<class 模板引數表> class 類名{ // 類定義...... };其中,template 是宣告類模板的關鍵字,表示宣告一個模板,模板引數可以是一個,也可以是多個,可以是型別引數,也可以是非型別引數。型

    求兩個整數的最大公約數和最小公倍數通過呼叫定義函式實現

    >#define _CRT_SECURE_NO_WARNINGS #include<stdio.h> int yue(int x, int y); //int yue_2(int

    SpringMVC框架11.3 定義引數繫結

    一、自定義引數繫結-屬性編輯器(不推薦) 問題:① 4.1 itemsList.jsp 中增加顯示 “訂購日期” 屬性;② JSP頁面中日期拿到的是字串,而提交到Controller中POJO類ItemsCustom 屬性物件的日期欄位要變成Date型別,即字串轉換成日期型別,無法自動轉

    spring security 5.x 使用及分析二:定義配置—初階

    二、自定義配置(初階): 自定義的配置,就要修改一些預設配置的資訊,從那開始入手呢? 1、第一步:建立Spring Security 的Java配置,改配置建立一個名為springSecurityFilterChain的servlet過濾器,它負責應用程式的安

    Oracle定義函式實現動態引數複製表使用了定義type以及pipelined

    (作者:陳玓玏) 之前試了一下,想用自定義函式及遊標實現動態傳入引數,確實可以,但是輸出結果總是不能成表格。 查了一圈,Oracle自定義函式好像是不能直接在SQL語句中寫create as select和insert into這些功能的,但是後來的版本中提供了

    HttpClient正確設定Host的姿勢不需要定義DNS解析類

    因為有負載均衡的考慮,前端用了nginx反向代理。 兩個域名雖然IP相同,但是如果不設定hosts檔案,直接通過IP,是會返回404的。 所以需求就是這樣,不設定hosts檔案,而正確訪問到對應的域名。 其實訪問的IP是一樣的,只是request header的Host不一

    caffefaster-RCNN環境搭建

    mit argument .com tail uil check otto pip __init__ faster-rcnn提出論文: 《Faster R-CNN: Towards Real-Time Object Detection with Region Propo

    七扭八歪解faster rcnnkeras

    def get_data(input_path): all_imgs = {} classes_count = {} class_mapping = {} with open(input_path,'r') as f: pr

    用於Unity(windows\iOS\安卓)的CLZMA壓縮演算法庫dll .so和.a

    人生苦短,道阻且艱;修行不易,且行且努力。 【專業擅長領域】:iOS開發,遊戲開發,圖形學 【擅長平臺】:iOS平臺,Unity --------------------------------------------------------- 【個人主頁】:信厚

    KerasFaster-RCNN程式碼學習IOU,RPN1

    config.py from keras import backend as K import math class Config: def __init__(self): self.verbose = True