mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-30 16:31:08 +00:00
daa1c7529a
* ci: Add `B023` to `.flake8` ignores * refactor: Return `bool`/raise Exception * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * revert: Remove previous branch commit * Update data_structures/binary_tree/segment_tree_other.py Co-authored-by: Christian Clauss <cclauss@me.com> * feat: Apply `__repr__` changes * chore: Fix failing tests * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update data_structures/binary_tree/segment_tree_other.py Co-authored-by: Christian Clauss <cclauss@me.com> * test: Fix doctests * random.choice(population_score[:N_SELECTED])[0] * Update basic_string.py Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Christian Clauss <cclauss@me.com>
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
import cv2
|
|
import numpy as np
|
|
|
|
"""
|
|
Harris Corner Detector
|
|
https://en.wikipedia.org/wiki/Harris_Corner_Detector
|
|
"""
|
|
|
|
|
|
class HarrisCorner:
|
|
def __init__(self, k: float, window_size: int):
|
|
|
|
"""
|
|
k : is an empirically determined constant in [0.04,0.06]
|
|
window_size : neighbourhoods considered
|
|
"""
|
|
|
|
if k in (0.04, 0.06):
|
|
self.k = k
|
|
self.window_size = window_size
|
|
else:
|
|
raise ValueError("invalid k value")
|
|
|
|
def __str__(self) -> str:
|
|
return str(self.k)
|
|
|
|
def detect(self, img_path: str) -> tuple[cv2.Mat, list[list[int]]]:
|
|
|
|
"""
|
|
Returns the image with corners identified
|
|
img_path : path of the image
|
|
output : list of the corner positions, image
|
|
"""
|
|
|
|
img = cv2.imread(img_path, 0)
|
|
h, w = img.shape
|
|
corner_list: list[list[int]] = []
|
|
color_img = img.copy()
|
|
color_img = cv2.cvtColor(color_img, cv2.COLOR_GRAY2RGB)
|
|
dy, dx = np.gradient(img)
|
|
ixx = dx**2
|
|
iyy = dy**2
|
|
ixy = dx * dy
|
|
k = 0.04
|
|
offset = self.window_size // 2
|
|
for y in range(offset, h - offset):
|
|
for x in range(offset, w - offset):
|
|
wxx = ixx[
|
|
y - offset : y + offset + 1, x - offset : x + offset + 1
|
|
].sum()
|
|
wyy = iyy[
|
|
y - offset : y + offset + 1, x - offset : x + offset + 1
|
|
].sum()
|
|
wxy = ixy[
|
|
y - offset : y + offset + 1, x - offset : x + offset + 1
|
|
].sum()
|
|
|
|
det = (wxx * wyy) - (wxy**2)
|
|
trace = wxx + wyy
|
|
r = det - k * (trace**2)
|
|
# Can change the value
|
|
if r > 0.5:
|
|
corner_list.append([x, y, r])
|
|
color_img.itemset((y, x, 0), 0)
|
|
color_img.itemset((y, x, 1), 0)
|
|
color_img.itemset((y, x, 2), 255)
|
|
return color_img, corner_list
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
edge_detect = HarrisCorner(0.04, 3)
|
|
color_img, _ = edge_detect.detect("path_to_image")
|
|
cv2.imwrite("detect.png", color_img)
|