def output_fn()

in code/inference.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

def output_fn(output_batch, accept='application/json'):
    res = []
    print('output list length')
    print(len(output_batch))
    for output in output_batch:
         res.append({'boxes':output['boxes'].detach().cpu().numpy().tolist(),'labels':output['labels'].detach().cpu().numpy().tolist(),'scores':output['scores'].detach().cpu().numpy().tolist()})
    
    return json.dumps(res)