算法生成的图片返回前端展示的简易方案

这篇博客介绍了如何在Python环境中运行模型,特别是计算机视觉(CV)模型,将模型输出的numpy数组转换为图片并以base64编码的方式在前端展示。通过使用BytesIO避免了文件落地,直接在内存中完成图片转JPEG格式和base64编码的过程,便于在HTML中嵌入显示。这种方法适用于算法工程师快速测试API,但可能不适用于需要图片缓存等生产环境需求。
摘要由CSDN通过智能技术生成

本文只针对python的环境
跑一个模型,特别是CV的模型,结果可能是一张图片,这张图片返回前端展示有多种方法。这里介绍其中一种方案,方便算法工程师测试API(不一定满足生产要求,比如图片需要在客户端缓存等,只是测试算法API应该是够用的)

模型生成的结果往往是numpy数组,或者opencv、PIL等库的图片格式。以numpy数组为例,首先需要把这个数据转换成.jpg等图片格式。为了避免落盘,可以把图片文件写入内存,使用BytesIO,写入内存后再读出,并使用base64编码。

im_show = Image.fromarray(im_show[:,:,[2,1,0
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值