heatmap

import numpy as np
import matplotlib as mpl
from PIL import Image, ImageFont, ImageDraw
import matplotlib.pyplot as plt
import cv2


img = Image.open("/ByteTrack-y/tools/save/bgimg.jpg")
# img = img.load()
img_rgb = img.convert("RGB")
img_array = np.array(img_rgb)
# print(img_array.shape)
w = img.size[0]
h = img.size[1]
# print(w)
# print(h)
img2_array = []
#pos表示修改像素点的位置，嵌套列表中第一个数值表示行号，第二个数值表示列号
pos=[[0,0],[180,190]]
for i in range(0, h):
    for j in range(0, w):
        temp = [i,j]
        color_list = []
        if temp in pos :
            img_array[i, j] = [255, 255, 255]
            cv2.circle(img_array, (i, j), 20, (255, 255, 255), -1)
            #[255, 0, 0]为红色，[255, 255, 255]为白色，[0, 0, 0]为黑色等
            img2_array.append(img_array[i, j])
        else:
            img2_array.append(img_array[i, j])

img2_array = np.asarray(img2_array)
print(img2_array)
img2_array = img2_array.reshape(480, 640, 3)
img3 = Image.fromarray(img2_array)
# img3.show()
plt.imshow(img3)
plt.show()
img3.save(r"/ByteTrack-y/tools/save/ss.jpg")