def predict_fn()

in src/inference_mxnet.py [0:0]

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

def predict_fn(input_object, model):
    '''
    Predicts on the deserialized object with the model from model_fn().
    '''
    x = input_object
    
    t0 = time.time()
    cid, score, bbox = model(x.as_in_context(ctx))
    t1 = time.time() - t0
    print("--- Elapsed time: %s secs ---" % t1)

    return x.shape, cid[0], score[0], bbox[0]