此外，計(jì)算bbox誤差時(shí)听怕，不是比較四個(gè)角的坐標(biāo)捧挺，而是tx，ty尿瞭，tw闽烙，th，具體計(jì)算如下：
預(yù)測的中心點(diǎn)和長寬值
明明只需要log声搁，? ? 卻除以了log（2）黑竞，轉(zhuǎn)換底，因?yàn)閏++里面的log是默認(rèn)以e為底疏旨，然而e不好打出來很魂，另外的? ?fit_ratio是為了讓值顯得大一些，才進(jìn)行的檐涝。
train中
void CHECK_IOU(Anchor<Dtype>& anchor, const std::vector<Box<Dtype>> >_Boxes){
int target_num = GT_Boxes.size();
Dtype area1 = anchor.width * anchor.height;
Dtype maxoverlap = 0.f;
int index = -1;
for (int i = 0; i < target_num; i++) {
const Box<Dtype> >_box = GT_Boxes[i];
Dtype x0 = std::max(anchor.x_ctr - anchor.width / 2, gt_box.x0);
Dtype y0 = std::max(anchor.y_ctr - anchor.height / 2, gt_box.y0);
Dtype x1 = std::min(anchor.x_ctr + anchor.width / 2, gt_box.x0 + gt_box.width);
Dtype y1 = std::min(anchor.y_ctr + anchor.height / 2, gt_box.y0 + gt_box.height);
Dtype area = (x1 - x0 + 1) * (y1 - y0 + 1);
if (x1 - x0 + 1 < 0 || y1 - y0 + 1 < 0){
continue;
}
Dtype area2 = gt_box.width * gt_box.height;
Dtype Overlap = area / (area1 + area2 - area);
if (Overlap > maxoverlap) {
maxoverlap = Overlap;
index = i;
}
}
anchor.overlap = maxoverlap;
if (maxoverlap > Part_overlap) {
const Box<Dtype> &box = GT_Boxes[index];
anchor.name = box.name;
if (maxoverlap > Pos_overlap) {
anchor.isobject = 2;
}
else {
anchor.isobject = 1;
}
anchor.Tar[0] = (box.x0 + box.width / 2 - anchor.x_ctr) / anchor.width;
anchor.Tar[1] = (box.y0 + box.height / 2 - anchor.y_ctr) / anchor.height;
anchor.Tar[2] = std::log(box.width / anchor.width) / std::log(2) * fit_ratio;
anchor.Tar[3] = std::log(box.height / anchor.height) / std::log(2) * fit_ratio;
}
else {
anchor.isobject = 0;
caffe::caffe_memset(4 * sizeof(Dtype), 0, anchor.Tar);
}
if (LabelIndex.count(anchor.name)) {
anchor.cls = LabelIndex.find(anchor.name)->second;
}
}
test中
微調(diào)box
template < typename Dtype >
inline void ReassignBox(Box &box, const Anchor<Dtype> &anchor,
Dtype x_ctr_off, Dtype y_ctr_off, Dtype w_off, Dtype h_off){
box.width = std::pow(2, w_off / fit_ratio) * anchor.width;
box.height = std::pow(2, h_off / fit_ratio) * anchor.height;
box.x0 = x_ctr_off * anchor.width + anchor.x_ctr - box.width / 2;
box.y0 = y_ctr_off * anchor.height + anchor.y_ctr - box.height / 2;
}
