core::rcnn Namespace Reference

Functions
def	get_rcnn_batch
def	get_rcnn_testbatch
def	sample_rois

---

Function Documentation

def core.rcnn.get_rcnn_batch	(		roidb,
			cfg
	)

return a dict of multiple images
:param roidb: a list of dict, whose length controls batch size
['images', 'flipped'] + ['gt_boxes', 'boxes', 'gt_overlap'] => ['bbox_targets']
:return: data, label

Definition at line 52 of file rcnn.py.

def core.rcnn.get_rcnn_testbatch	(		roidb,
			cfg
	)

return a dict of testbatch
:param roidb: ['image', 'flipped'] + ['boxes']
:return: data, label, im_info

Definition at line 30 of file rcnn.py.

def core.rcnn.sample_rois	(		rois,
			fg_rois_per_image,
			rois_per_image,
			num_classes,
			cfg,
			labels = None,
			overlaps = None,
			bbox_targets = None,
			gt_boxes = None
	)

generate random sample of ROIs comprising foreground and background examples
:param rois: all_rois [n, 4]; e2e: [n, 5] with batch_index
:param fg_rois_per_image: foreground roi number
:param rois_per_image: total roi number
:param num_classes: number of classes
:param labels: maybe precomputed
:param overlaps: maybe precomputed (max_overlaps)
:param bbox_targets: maybe precomputed
:param gt_boxes: optional for e2e [n, 5] (x1, y1, x2, y2, cls)
:return: (labels, rois, bbox_targets, bbox_weights)

Definition at line 120 of file rcnn.py.