Add histograms

This commit is contained in:
Andras Schmelczer 2024-04-05 21:47:18 +01:00
parent c7a2890e99
commit c0b0dacd99
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
3 changed files with 34738 additions and 0 deletions

View file

@ -0,0 +1,14 @@
from PIL import Image
import numpy as np
def compute_histogram(image_path, bins: int, value_range=(0, 256)):
image = Image.open(image_path)
image = np.array(image)
histogram, _ = np.histogramdd(
image.reshape(-1, 3), bins=bins, range=[value_range, value_range, value_range]
)
histogram = histogram / np.sum(histogram)
return histogram.astype(np.float32)