bbox::bbox_regression Namespace Reference

Functions
def	add_bbox_regression_targets
def	compute_bbox_regression_targets
def	expand_bbox_regression_targets

---

Function Documentation

def bbox.bbox_regression.add_bbox_regression_targets	(		roidb,
			cfg
	)

given roidb, add ['bbox_targets'] and normalize bounding box regression targets
:param roidb: roidb to be processed. must have gone through imdb.prepare_roidb
:return: means, std variances of targets

Definition at line 56 of file bbox_regression.py.

def bbox.bbox_regression.compute_bbox_regression_targets	(		rois,
			overlaps,
			labels,
			cfg
	)

given rois, overlaps, gt labels, compute bounding box regression targets
:param rois: roidb[i]['boxes'] k * 4
:param overlaps: roidb[i]['max_overlaps'] k * 1
:param labels: roidb[i]['max_classes'] k * 1
:return: targets[i][class, dx, dy, dw, dh] k * 5

Definition at line 19 of file bbox_regression.py.

def bbox.bbox_regression.expand_bbox_regression_targets	(		bbox_targets_data,
			num_classes,
			cfg
	)

expand from 5 to 4 * num_classes; only the right class has non-zero bbox regression targets
:param bbox_targets_data: [k * 5]
:param num_classes: number of classes
:return: bbox target processed [k * 4 num_classes]
bbox_weights ! only foreground boxes have bbox regression computation!

Definition at line 116 of file bbox_regression.py.