将matplotlib绘制的fig保存在缓存中,用于cv2读取

做检测避免不了使用matplotlib和cv2这些库,但是图片在这两个库里面的格式是不一样的。cv2里面图片是numpy array,但是matplotlib就不是,cv2里颜色顺序是BGR,而matplotlib是RGB。因为经常需要处理摄像头画面和视频流,查了一些关于这方面的转换方法,我试了下都遇到了各种各样的问题,所以为了避免麻烦,我采用的思路是将matplotlib绘图的结果保存,再用cv2读取并显示。如果保存在本地磁盘可能会导致速度慢,查到可以将结果保存在内存中,说是这样会提高些效率。

python3

import io
import numpy as np
import PIL
import matplotlib.pyplot as plt

#... fig = ...
#假设你有了一个绘图结果fig

#申请缓存
buffer_ = io.BytesIO()
fig.savefig(buffer_, format = "png")
buffer_.seek(0)
image = PIL.Image.open(buffer_)
#转换为numpy array
ar = np.asarray(image)
cv2.imshow("demo", ar)
#释放缓存
buffer_.close()

  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值