mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-27 15:01:08 +00:00
Hacktoberfest 2020: Added computer vision algorithm (#2946)
* Create meanthresholding.py * Rename meanthresholding.py to meanthreshold.py * Update meanthreshold.py * Update computer_vision/meanthreshold.py Verified this part works, thanks. Co-authored-by: Christian Clauss <cclauss@me.com> * Update computer_vision/meanthreshold.py Co-authored-by: Christian Clauss <cclauss@me.com> Co-authored-by: Christian Clauss <cclauss@me.com>
This commit is contained in:
parent
c33b683193
commit
fc98961814
30
computer_vision/meanthreshold.py
Normal file
30
computer_vision/meanthreshold.py
Normal file
|
@ -0,0 +1,30 @@
|
||||||
|
from PIL import Image
|
||||||
|
|
||||||
|
"""
|
||||||
|
Mean thresholding algorithm for image processing
|
||||||
|
https://en.wikipedia.org/wiki/Thresholding_(image_processing)
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
def mean_threshold(image: Image) -> Image:
|
||||||
|
"""
|
||||||
|
image: is a grayscale PIL image object
|
||||||
|
"""
|
||||||
|
height, width = image.size
|
||||||
|
mean = 0
|
||||||
|
pixels = image.load()
|
||||||
|
for i in range(width):
|
||||||
|
for j in range(height):
|
||||||
|
pixel = pixels[j, i]
|
||||||
|
mean += pixel
|
||||||
|
mean //= width * height
|
||||||
|
|
||||||
|
for j in range(width):
|
||||||
|
for i in range(height):
|
||||||
|
pixels[i, j] = 255 if pixels[i, j] > mean else 0
|
||||||
|
return image
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
image = mean_threshold(Image.open("path_to_image").convert("L"))
|
||||||
|
image.save("output_image_path")
|
Loading…
Reference in New Issue
Block a user