mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2024-11-23 21:11:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

[mypy] annotate computer_vision (#5571)

Browse Source
This commit is contained in:
Erwin Junge 2021-10-23 23:26:21 +02:00 committed by GitHub
parent 00a67010e8
commit aaaa175b66
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
computer_vision/harris_corner.py
View File

@ -21,11 +21,11 @@ class Harris_Corner:
else:
raise ValueError("invalid k value")
def __str__(self):
def __str__(self) -> str:
return f"Harris Corner detection with k : {self.k}"
def detect(self, img_path: str):
def detect(self, img_path: str) -> tuple[cv2.Mat, list[list[int]]]:
"""
Returns the image with corners identified
@ -35,7 +35,7 @@ class Harris_Corner:
img = cv2.imread(img_path, 0)
h, w = img.shape
corner_list = []
corner_list: list[list[int]] = []
color_img = img.copy()
color_img = cv2.cvtColor(color_img, cv2.COLOR_GRAY2RGB)
dy, dx = np.gradient(img)