def predict()

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

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

def predict(filename, network):
    x, img = data.transforms.presets.ssd.load_test(filename, short=512)
    class_IDs, scores, bounding_boxes = net(x)

    return class_IDs, scores, bounding_boxes