caffe原始碼分析 IoU計算

2021-09-11 07:39:32 字數 900 閱讀 6106

iou

=pre

dict

檢測框與

grou

nd_t

ruth

檢測框的

交集pr

edic

t檢測框

與gro

und_

trut

h檢測框

的的並集

iou = \frac

iou=pr

edic

t檢測框

與gro

und_

trut

h檢測框

的的並集

pred

ict檢

測框與g

roun

d_tr

uth檢

測框的交

集​iou就是caffe.ssd中的jaccardoverlap。

計算iou

來自caffe.ssd/src/caffe/util/bbox_util.cpp

template dtype jaccardoverlap(const dtype* bbox1, const dtype* bbox2)  else 

}

求bbox面積

來自caffe.ssd/src/caffe/util/bbox_util.cpp

template dtype bboxsize(const dtype* bbox, const bool normalized = true);

template dtype bboxsize(const dtype* bbox, const bool normalized) else else

}}

caffe原始碼分析 DataTransformer

將datum型別或者cv mat,轉化為caffe的blob,並按照transformation parameter引數對影象做處理,例如scale,mirro等 推斷blob的shape proto定義如下 to the data layer s data message transformati...

Caffe引數交換原始碼分析

對境準備 對於多個gpu而言,一台機器2個gpu,引數交換的流程圖 引數交換從main 進入train 函式,在train函式中找到對應原始碼為 1 2if gpus.size 1 else 因為gpu的個數 1,所以執行sync solver,null,solver param 和run 函式,首...

Caffe原始碼分析隨筆 二 Blob

blob是caffe中每一層的基本型別。檢視標頭檔案 template class blob protected shared ptr data shared ptr diff shared ptr shape data vector shape int count int capacity blo...