heatmap

最新推荐文章于 2024-10-03 09:02:12 发布

水门汀汀

最新推荐文章于 2024-10-03 09:02:12 发布

阅读量94

点赞数

文章标签： python

本文链接：https://blog.csdn.net/weixin_51082922/article/details/121291463

版权

import numpy as np
import cv2

filepath = '/ByteTrack-y/tools/datasets/mot/train/MOT17-05-FRCNN/gt/gt.txt'
file_list = []
with open(filepath, 'r') as fp:
    for eachline in fp.readlines():
        eachline = eachline.rstrip().split(',')
        file_list.append(eachline)
        # print(file_list)
results = []
N = len(file_list)
for i in range(N-1):
    if file_list[i][1] != file_list[i+1][1] or file_list[i][1] != file_list[i-1][1]:
        results.append(file_list[i])
results.append(file_list[N-1])
# count = 0
# count1 = 0
bgimg = np.zeros((480, 640, 3), dtype=np.uint8)
for l in results:
    # print(l)

    l0, l1, l2, l3 = int(l[2]), int(l[3]), int(int(l[2])+int(l[4])), int(int(l[3])+int(l[5]))
    l0 = max(l0, 0)
    l1 = max(l1, 0)
    l2 = max(l2, 0)
    l3 = max(l3, 0)
    bgimg[l1:l3, l0:l2] += 1
    # count += (l2 <= 640)
    # print(l2)
    # count1 += (l0 >= 0)
# print(count)
# print(count1)
# print(len(results))
# print(np.max(bgimg[:, :430, :]))
# print(np.max(bgimg[:, 430:, :]))

cv2.normalize(bgimg, bgimg, 0, 255, cv2.NORM_MINMAX)
# bgimg[:, 200, :] = 255
new_bgimg = cv2.imread("/ByteTrack-y/tools/datasets/mot/train/MOT17-05-FRCNN/img1/000001.jpg")
# print(new_bgimg.shape)
new_bgimg[:, :, 2] = bgimg[:, :, 0]
# cv2.imshow("dd", bgimg)
# cv2.waitKey(0)
cv2.imwrite('/ByteTrack-y/tools/datasets/mot/train/MOT17-05-FRCNN/gt/sss.jpg', bgimg)
cv2.imwrite('/ByteTrack-y/tools/datasets/mot/train/MOT17-05-FRCNN/gt/sss2.jpg', new_bgimg)

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


img = Image.open("/home/wangliuan/intern/yuxi/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"/home/wangliuan/intern/yuxi/ByteTrack-y/tools/save/ss.jpg")