原文:MMDetection源碼解析:Faster RCNN(8)--BBoxHead類

BBoxHead類繼承自nn.Module類,定義在 mmdet models roi heads bbox heads bbox head.py中,其作用是輸出ROI Pooling的分類和回歸值. 主要的函數有: init :初始化函數,主要參數包括POI Pooling的尺寸大小,輸入通道數等等 get targets :計算目標值,通過調用 get target single 實現 get ...

2021-09-24 16:16 0 131 推薦指數:

查看詳情

MMDetection源碼解析Faster RCNN(2)--FasterRCNN

Faster RCNN配置文件faster_rcnn_r50_fpn.py中的 說明了這是一個Faster RCNN模型,對應的是mmdetection/mmdet/models/detectors/faster_rcnn.py文件,主要內容如下: 由代碼可知 ...

Tue Mar 30 01:43:00 CST 2021 0 258
MMDetection源碼解析Faster RCNN(3)--RPN Head

Faster RCNN配置文件faster_rcnn_r50_fpn.py中的 設置了Backbone為ResNet. 設置了Neck為FPN.Backbone和Neck比較簡單,就不詳細介紹了,詳細介紹一下RPN Head. 以上文件設置了RPN ...

Thu Apr 15 01:22:00 CST 2021 0 383
MMDetection源碼解析Faster RCNN(1)--配置文件

faster_rcnn_r50_fpn_1x_coco.py文件位於目錄mmdetection/configs/faster_rcnn/下面,主要內容如下: 其中第1個配置文件說明了使用哪一個模型,第2個配置文件說明了訓練的數據集,第3個配置文件說明了訓練的參數,如學習率 ...

Sun Feb 28 07:55:00 CST 2021 0 519
mmdetection faster-rcnn yolo

mmdetection安裝過程中依靠https://github.com/open-mmlab/mmdetection/blob/master/docs/get_started.md 然后在安裝第三步Install mmcv-full時,發現自己的cuda是10.1的,然后pytorch ...

Thu Feb 25 11:08:00 CST 2021 1 598
py-faster-rcnn 源碼解析1

faster rcnn是在fast rcnn的基礎上,包裝了RPN以及網絡共享。首先,看一下除fast rcnn以外的內容。 tools 1._init_paths.py 初始化路徑,要訪問module/package的路徑的加進去,給PYTHONPATH加上訪問caffe和lib的路徑 ...

Wed Oct 25 07:05:00 CST 2017 0 1156
Faster RCNN代碼解析

1.faster_rcnn_end2end訓練 1.1訓練入口及配置 1.2 數據准備 從train_net.py:combined_roidb(imdb_name)處開始,得到的是gt數據集。 輸入:“voc_2007_trainval ...

Thu Nov 01 01:22:00 CST 2018 0 2718
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM