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

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

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

im_show = Image.fromarray(im_show[:,:,[2,1,0]]) # im_show是numpy数组,是模型的返回值
f = BytesIO()
im_show.save(f, 'jpeg')
data = f.getvalue()
data = base64.b64encode(data).decode()

各种web框架都是支持返回html的,所以只需要把上述base64码嵌入html就ok了,这样就可以在客户端方便地查看模型返回的结果了。

'<!DOCTYPE html><html lang="en"><meta charset="UTF-8"><div><img src="https://img-blog.csdnimg.cn/2022010705335626754.png"/></div>'.format(data)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值