def calc_histo(image, hsize, ranges=[0, 256]):
hist = np.zeros((hsize, 1), np.float32)
gap = ranges[1] / hsize
for i in range(image.shape[0]):
for j in range(image.shape[1]):
idx = int(image.item(i,j) / gap)
hist[idx] += 1
return hist
위 함수를 사용해서 아래와 같은 2차원 히스토그램 분석 결과를 얻고싶은데 방법을 모르겠어요.
[[1.0234e+04 1.1613e+04 6.9510e+03 0.0000e+00]
[3.7950e+03 6.2410e+03 3.1410e+03 6.0000e+00]
[0.0000e+00 1.5290e+03 1.3261e+04 2.1480e+03]
[0.0000e+00 0.0000e+00 4.1800e+02 1.2663e+04]]
제가 작성한 코드는
import numpy as np, cv2
def calc_histo(image, hsize, ranges=[0, 256]):
hist = np.zeros((hsize, 1), np.float32)
gap = ranges[1] / hsize
for i in range(image.shape[0]):
for j in range(image.shape[1]):
idx = int(image.item(i,j) / gap)
hist[idx] += 1
return hist
image = cv2.imread("images/pixel.jpg", cv2.IMREAD_COLOR)
if image is None:
raise Exception("영상 파일 읽기 오류 발생")
ch, bins, ranges = [0, 1], [4, 4], [0, 256, 0, 256]
image2 = np.zeros((image.shape[0], 256), dtype=np.float32)
for i in bins:
hist1 = calc_histo(image2, i, ranges)
hist2 = cv2.calcHist([image], ch, None, bins, ranges)
print("저자 구현 함수: \n", hist1)
print("OpenCV 함수: \n", hist2)
cv2.imshow("image", image)
cv2.waitKey(0)
이것이고 결과는
저자 구현 함수:
[[76800.]
[ 0.]
[ 0.]
[ 0.]]
OpenCV 함수:
[[1.0234e+04 1.1613e+04 6.9510e+03 0.0000e+00]
[3.7950e+03 6.2410e+03 3.1410e+03 6.0000e+00]
[0.0000e+00 1.5290e+03 1.3261e+04 2.1480e+03]
[0.0000e+00 0.0000e+00 4.1800e+02 1.2663e+04]]
이렇게 나옵니다.