fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-05-13 01:18:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

chg: [mypy] fixed mypy issue

Browse source
This commit is contained in:
Cédric Bonhomme 2021-11-24 12:11:26 +01:00
parent 27083934d0
commit 0460acd7f3
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
stegano/lsbset/generators.py
View file

@ -29,7 +29,7 @@ import itertools
import cv2 import cv2
import numpy as np import numpy as np
import math import math
from typing import Dict, Iterator, List from typing import Dict, Iterator, List, Any, Union
def identity() -> Iterator[int]: def identity() -> Iterator[int]:
@ -230,14 +230,15 @@ def LFSR(m: int) -> Iterator[int]:
def shi_tomashi( def shi_tomashi(
image: str, corners: int = 100, quality: float = 0.01, min_distance: float = 10.0 image_path: str, corners: int = 100, quality: float = 0.01, min_distance: float = 10.0
) -> Iterator[int]: ) -> Iterator[int]:
"""Shi-Tomachi corner generator of the given points """Shi-Tomachi corner generator of the given points
https://docs.opencv.org/4.x/d4/d8c/tutorial_py_shi_tomasi.html https://docs.opencv.org/4.x/d4/d8c/tutorial_py_shi_tomasi.html
""" """
image = cv2.imread(image) image = cv2.imread(image_path)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
corners = cv2.goodFeaturesToTrack(gray, corners, quality, min_distance) corners: Union[int, List[Any]] = cv2.goodFeaturesToTrack(
gray, corners, quality, min_distance)
corners = np.int0(corners) corners = np.int0(corners)
i = 0 i = 0
while True: while True: