在原图上绘制预测的关键点热图

最新推荐文章于 2024-07-18 15:53:23 发布

Ctrl_Cver

最新推荐文章于 2024-07-18 15:53:23 发布

阅读量129

点赞数 3

文章标签：人工智能

本文链接：https://blog.csdn.net/qq_33596242/article/details/137020671

版权

import os
import cv2
import numpy as np

img = cv2.imread('/Users/tfj/Downloads/1.jpg')  # 图片里是crop出来的人，hw(256,192)
hms = np.load('/Users/tfj/Downloads/person_heatmap.npy')  # 网络的预测热图，(17,64,48)

# 保存17张图
iii=0
for hm in hms:
    rs_hm = cv2.resize(hm,(192,256))
    rs_hm = np.clip(rs_hm*255,0,255)
    rs_hm = np.array(rs_hm,dtype=np.uint8)
    colored_heatmap = cv2.applyColorMap(rs_hm, cv2.COLORMAP_JET)
    masked_image = colored_heatmap*0.5 + img*0.5

    cv2.imwrite(str(iii)+'.jpg',masked_image)
    iii+=1