我有一个Torch张量,我需要将其转换为Byte对象,以便将其传递给Starlette的StreamingResponse
,后者将从Byte对象返回重建的图像.我正在try 转换张量并返回它,如下所示:
def some_unimportant_function(params):
return_image = io.BytesIO()
torch.save(some_img, return_image)
return_image.seek(0)
return_img = return_image.read()
return StreamingResponse(content=return_img, media_type="image/jpeg")
下面的代码可以很好地处理常规的Byte对象,并且我的API将返回重建的图像:
def some_unimportant_function(params):
image = Image.open(io.BytesIO(some_image))
return_image = io.BytesIO()
image.save(return_image, "JPEG")
return_image.seek(0)
return StreamingResponse(content=return_image, media_type="image/jpeg")
为此使用PIL
个库
我到底做错了什么?