283375/arcaea-offline-ocr
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline-ocr.git synced 2025-04-19 05:20:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

impr(sift_db): minor improvements

Browse Source
This commit is contained in:
283375 2023-08-16 18:13:31 +08:00
parent 42979c67cb
commit 5fc8581a3f
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/arcaea_offline_ocr/sift_db.py
View File

@ -16,6 +16,9 @@ class SIFTDatabase:
self.__descriptors = [] self.__descriptors = []
self.__size = None self.__size = None
self.__sift = cv2.SIFT_create()
self.__bf_matcher = cv2.BFMatcher()
if load: if load:
self.load_db() self.load_db()
@ -43,6 +46,14 @@ class SIFTDatabase:
def size(self, value: Tuple[int, int]): def size(self, value: Tuple[int, int]):
self.__size = value self.__size = value
@property
def sift(self):
return self.__sift
@property
def bf_matcher(self):
return self.__bf_matcher
def load_db(self): def load_db(self):
conn = sqlite3.connect(self.db_path) conn = sqlite3.connect(self.db_path)
with conn: with conn:
@ -75,9 +86,12 @@ class SIFTDatabase:
self, self,
__img: Mat, __img: Mat,
*, *,
sift=cv2.SIFT_create(), sift=None,
bf: cv2.BFMatcher = cv2.BFMatcher(), bf=None,
) -> Tuple[str, float]: ) -> Tuple[str, float]:
sift = sift or self.sift
bf = bf or self.bf_matcher
img = __img.copy() img = __img.copy()
if self.size is not None: if self.size is not None:
img = cv2.resize(img, self.size) img = cv2.resize(img, self.size)