def d2_to_json()

in source/d2_deserializer.py [0:0]

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

def d2_to_json(predictions):
    """
    Server side helper function to serialize the d2 detections into JSON for API passing 
    """
    
    instances = predictions["instances"]
    output = {}

    # Iterate over fields in Instances
    for k,v in instances.get_fields().items():
        
        if k in ["scores", "pred_classes"]:
            output[k] = v.tolist()
            
        if k=="pred_boxes":
            output[k] = v.tensor.tolist()
            
    output['image_size'] = instances.image_size
    output = json.dumps(output)
    
    return output