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

版权声明：原创文章，转载请标明来源 https://blog.csdn.net/ngy321/article/details/80109088

做检测避免不了使用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()