如何将TkImage原始数据直接传入OpenCV（cv2）函数？

我正在使用tkinter和cv2编写视频录制程序，如何将TkImage原始数据传入cv2.VideoWriter、cv2.circle等函数？

目前使用numpy数组（self.data）作为中间载体，但这种方式每次都需要重新创建TkImage，希望在创建一次TkImage后，直接用cv2函数访问其数据。以下是封装了TkImage和numpy数组的类：

import cv2
import numpy as np
# import tkinter as tk
# from tkinter import Canvas
from PIL import Image, ImageTk

class Bitmap:
    def __init__(self, size=(512, 512)):
        self.data = np.zeros((size[1], size[0], 3), dtype=np.uint8)
        self._image_id = 0
        self._image = None

    @property
    def size(self):
        return (self.data.shape[1], self.data.shape[0])

    @property
    def width(self):
        return self.data.shape[1]

    @property
    def height(self):
        return self.data.shape[0]

    @property
    def clear(self):
        self.data.fill(0)

    def new_image(self):
        self._image = ImageTk.PhotoImage(image=Image.fromarray(self.data))
        return self._image_id, self._image

    @property
    def image(self):
        return self._image_id, self._image

    def setup(self, canvas):
        self._image = ImageTk.PhotoImage(image=Image.fromarray(self.data))
        self._image_id = canvas.create_image(0, 0, anchor=tk.NW, image=self._image)

写入self.data示例：

cv2.circle(self.bitmap.data, (x, y), 5, (255, 255, 255), 1, cv2.LINE_AA)

写入视频文件示例：

# self.recorder = cv2.VideoWriter()
self.recorder.write(self.bitmap.data)

使用self.image更新tkinter界面示例：

# self._image = ImageTk.PhotoImage(image=Image.fromarray(self.data))
id, image = self.bitmap.image
self.canvas.itemconfig(id, image=image)

无法直接将TkImage的原始数据传入cv2函数——无论是tkinter原生的PhotoImage还是ImageTk.PhotoImage，都没有提供可直接被cv2识别的原始像素数据访问接口，也无法直接转换为cv2依赖的numpy数组格式。

最优方案是保留numpy数组作为核心数据源，让cv2直接操作numpy数组，同时优化TkImage的更新逻辑，避免不必要的实例重建：

优化后的Bitmap类 #

import cv2
import numpy as np
import tkinter as tk
from tkinter import Canvas
from PIL import Image, ImageTk

class Bitmap:
    def __init__(self, size=(512, 512)):
        # 核心数据源：numpy数组，cv2直接操作它
        self.data = np.zeros((size[1], size[0], 3), dtype=np.uint8)
        # 维护PIL Image和TkImage实例
        self._pil_image = Image.fromarray(self.data)
        self._image = ImageTk.PhotoImage(image=self._pil_image)
        self._image_id = 0
        # 标记数据是否修改，避免重复更新TkImage
        self._dirty = False

    @property
    def size(self):
        return (self.data.shape[1], self.data.shape[0])

    @property
    def width(self):
        return self.data.shape[1]

    @property
    def height(self):
        return self.data.shape[0]

    def clear(self):
        self.data.fill(0)
        self._dirty = True

    def mark_dirty(self):
        # 修改数据后标记为脏，触发后续界面更新
        self._dirty = True

    def get_updated_image(self):
        # 仅当数据修改时才重新创建TkImage
        if self._dirty:
            self._pil_image = Image.fromarray(self.data)
            self._image = ImageTk.PhotoImage(image=self._pil_image)
            self._dirty = False
        return self._image_id, self._image

    def setup(self, canvas):
        self._image_id = canvas.create_image(0, 0, anchor=tk.NW, image=self._image)

使用示例 #

1. 用cv2修改图形：

cv2.circle(self.bitmap.data, (x, y), 5, (255, 255, 255), 1, cv2.LINE_AA)
# 标记数据已修改
self.bitmap.mark_dirty()

2. 写入视频：

# 直接写入numpy数组，无需转换
self.recorder.write(self.bitmap.data)

3. 更新tkinter界面：

# 仅当数据修改时才更新TkImage
img_id, img = self.bitmap.get_updated_image()
self.canvas.itemconfig(img_id, image=img)

这种方式既保留了cv2操作numpy数组的高效性，又避免了不必要的TkImage重建，平衡了性能和开发复杂度。

内容的提问来源于stack exchange，提问作者cpicanco