minor improvements
This commit is contained in:
parent
6353ae7f78
commit
703019c4cd
8 changed files with 123 additions and 214 deletions
|
|
@ -3,9 +3,12 @@ import numpy as np
|
|||
|
||||
|
||||
def compute_histogram(
|
||||
image: Image, bins: int, value_range=(0, 256), normalize: bool = True
|
||||
):
|
||||
image = np.array(image)
|
||||
image: Image.Image | np.ndarray,
|
||||
bins: int,
|
||||
value_range=(0, 256),
|
||||
normalize: bool = True,
|
||||
) -> np.ndarray:
|
||||
image = np.array(image) if isinstance(image, Image.Image) else image
|
||||
|
||||
histogram, _ = np.histogramdd(
|
||||
image.reshape(-1, 3), bins=bins, range=[value_range, value_range, value_range]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue