INNER CODE UNIT · Python

get_segmentation_map

davidefiocco/streamlit-fastapi-model-serving · fastapi/server.py:19

def get_segmentation_map(file: bytes = File(...)):
    """Get segmentation maps from image file"""
    segmented_image = get_segments(model, file)
    bytes_io = io.BytesIO()
    segmented_image.save(bytes_io, format="PNG")
    return Response(bytes_io.getvalue(), media_type="image/png")

View source record →