OpenCV Python을 사용하여 HeatMap 흉내내기 – applyColorMap 사용

applyColorMap를 사용하여 OpenCV Python에서 HeatMap을 흉내내는 예제코드입니다.

 

2023. 9. 10 최초작성

 

배열에 가상으로 값을 넣어서 출력해본 결과입니다. 배열에 있는 값이 높을 수록 검은색 < 빨간색 < 노란색으로 보입니다.

 

전체 코드입니다.

 

import cv2 import numpy as np # 가상으로 점유 정보가 저장된 2차원 배열을 생성합니다. # OpenCV의 컬러맵 함수는 8비트 이미지를 사용하므로 0-255 범위로 설정합니다. 필요시 값의 범위를 조정해야합니다. occupancy_counts = np.array([[0,0,0,0,0,0,0,0,0,0],                             [0,10,10,10,10,10,10,10,10,0],                             [0,0,0,0,0,0,0,0,110,0],                             [0,0,0,0,0,0,0,0,110,0],                             [0,130,240,240,240,240,240,0,110,0],                             [0,130,0,0,255,0,250,0,110,0],                             [0,130,0,0,250,250,250,0,110,0],                             [0,130,0,0,0,0,0,0,110,0],                             [0,210,210,210,210,210,210,210,10,0],                             [0,0,0,0,0,0,0,0,0,0],                             ], dtype=np.uint8) # 크기 조정을 합니다. occupancy_counts = cv2.resize(occupancy_counts, None, fx=20, fy=20) # 배열을 컬러맵을 사용하여 시각화합니다. heatmap = cv2.applyColorMap(occupancy_counts, cv2.COLORMAP_HOT) # 결과를 출력합니다. cv2.imshow('Occupancy Grid', heatmap) cv2.waitKey(0) cv2.destroyAllWindows()