如何使用 Python 堆实现图像处理工具包？

import heapq
import cv2

# 读取图片
img = cv2.imread("image.jpg")

# 转化为灰度图
gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 将像素值存储到堆中
heap = []
for i in range(gray_img.shape[0]):
    for j in range(gray_img.shape[1]):
        heapq.heappush(heap, gray_img[i][j])

# 从堆中取出像素值进行排序
sorted_values = []
while heap:
    sorted_values.append(heapq.heappop(heap))

# 将排序后的像素值写回图像中
for i in range(gray_img.shape[0]):
    for j in range(gray_img.shape[1]):
        gray_img[i][j] = sorted_values.pop(0)

# 显示图像
cv2.imshow("sorted image", gray_img)
cv2.waitKey(0)
cv2.destroyAllWindows()