arne/EV5_Beeldherk_Bomen
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added statistics

Browse Source
This commit is contained in:
Tom Selier 2023-09-24 10:26:19 +02:00
parent 387cdb1247
commit 5fae9c9a0d
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/experiments/algorithms/colourspace.py
View File

@ -2,18 +2,25 @@ import cv2
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import numpy as np import numpy as np
def imgStats(img):
mean = np.zeros(3)
std = np.zeros(3)
for i in range(img.shape[2]):
mean[i] = np.mean(hsv_img[:, :, i])
std[i] = np.std(hsv_img[:, :, i])
return mean, std
DATASET_PATH = "C:\\Users\\Tom\\Desktop\\Files\\Repositories\\snake-vault\\TREES\\dataset\\" DATASET_PATH = "C:\\Users\\Tom\\Desktop\\Files\\Repositories\\snake-vault\\TREES\\dataset\\"
full_img = cv2.imread(DATASET_PATH + "alder\\1.jpg", 1) full_img = cv2.imread(DATASET_PATH + "alder\\1.jpg", 1)
bgr_img = cv2.resize(full_img, (0, 0), fx=0.25, fy=0.25) bgr_img = cv2.resize(full_img, (0, 0), fx=0.25, fy=0.25)
hsv_img = cv2.cvtColor(bgr_img, cv2.COLOR_BGR2HSV) hsv_img = cv2.cvtColor(bgr_img, cv2.COLOR_BGR2HSV)
cv2.imshow('BGR',bgr_img) # cv2.imshow('BGR',bgr_img)
cv2.imshow('HSV',hsv_img) # cv2.imshow('HSV',hsv_img)
cv2.imshow('HSV - H',hsv_img[:, :, 0]) # cv2.imshow('HSV - H',hsv_img[:, :, 0])
cv2.imshow('HSV - S',hsv_img[:, :, 1]) # cv2.imshow('HSV - S',hsv_img[:, :, 1])
cv2.imshow('HSV - V',hsv_img[:, :, 2]) # cv2.imshow('HSV - V',hsv_img[:, :, 2])
hist = cv2.calcHist([hsv_img], [0], None, [256], [0,256]) hist = cv2.calcHist([hsv_img], [0], None, [256], [0,256])
plt.plot(hist, label='H') plt.plot(hist, label='H')
@ -21,6 +28,11 @@ hist = cv2.calcHist([hsv_img], [1], None, [256], [0,256])
plt.plot(hist, label='S') plt.plot(hist, label='S')
hist = cv2.calcHist([hsv_img], [2], None, [256], [0,256]) hist = cv2.calcHist([hsv_img], [2], None, [256], [0,256])
plt.plot(hist, label='V') plt.plot(hist, label='V')
mean, std = imgStats(hsv_img)
print(mean)
print(std)
plt.grid() plt.grid()
plt.legend() plt.legend()
plt.show() plt.show()