babysister.detector module

class babysister.detector.Detector(core_detector)

Bases: object

Additional logics for OBJ Detection.

Parameters:core_detector (YOLOv3) – Core detector.

detect(im, valid_classes, max_bb_size_ratio)

Wrapper for self.core_detector.detect() with additional logics.

Additional logics:

Only detect classes in valid_classes, omit others. Fiter boxes with size ratio greater than max_bb_size_ratio.

Parameters:

• im (ndarray) – input images.
• valid_classes (list of str) – Only detect these classes.
• max_bb_size_ratio (list of 2 int) – Maximum box width ratio, height ratio wrt im size.

Returns:

boxes (ndarray) is boxes coordinate in format [[x0, y0, x1, y1], …]. scores (ndarray) is confidence scores. labels (ndarray) is label indexes.

Return type:

[boxes, scores, labels]