def get_object_boxes()

in artifacts/count_people/infer.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

def get_object_boxes(network, class_ids, scores, bounding_boxes, object_label, threshold=threshold):
    good_scores = (scores[0,:,0] > threshold)
    good_classes = (class_ids[0,:,0] == network.classes.index(object_label))

    boxes = bounding_boxes[0,:,:].asnumpy()[(good_scores.asnumpy()*good_classes.asnumpy()) > 0]

    return boxes