将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()

要在PyQt5设计的GUI界面显示读取文件matplotlib绘制的图形,可以采用以下步骤: 1. 导入必要的库 ``` import sys from PyQt5.QtWidgets import QApplication, QMainWindow, QVBoxLayout, QWidget from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas from matplotlib.figure import Figure ``` 2. 创建一个绘图函数,以便将读取的数据绘制成图形 ``` def plot_data(data): fig = Figure(figsize=(5, 4), dpi=100) ax = fig.add_subplot(111) ax.plot(data) return fig ``` 3. 在主窗口创建一个QWidget,将绘图控件添加到其,并设置布局 ``` class MainWindow(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle('My Application') self.setGeometry(100, 100, 800, 600) central_widget = QWidget(self) self.setCentralWidget(central_widget) layout = QVBoxLayout(central_widget) canvas = FigureCanvas(plot_data(data)) layout.addWidget(canvas) ``` 4. 运行程序,即可在GUI界面显示读取文件matplotlib绘制的图形。 完整代码如下: ``` import sys from PyQt5.QtWidgets import QApplication, QMainWindow, QVBoxLayout, QWidget from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas from matplotlib.figure import Figure def plot_data(data): fig = Figure(figsize=(5, 4), dpi=100) ax = fig.add_subplot(111) ax.plot(data) return fig class MainWindow(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle('My Application') self.setGeometry(100, 100, 800, 600) central_widget = QWidget(self) self.setCentralWidget(central_widget) layout = QVBoxLayout(central_widget) canvas = FigureCanvas(plot_data(data)) layout.addWidget(canvas) if __name__ == '__main__': app = QApplication(sys.argv) window = MainWindow() window.show() sys.exit(app.exec_()) ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值