mirror of
https://github.com/283375/arcaea-offline-ocr.git
synced 2025-04-18 21:10:17 +00:00
impr(sift_db): minor improvements
This commit is contained in:
parent
42979c67cb
commit
5fc8581a3f
@ -16,6 +16,9 @@ class SIFTDatabase:
|
||||
self.__descriptors = []
|
||||
self.__size = None
|
||||
|
||||
self.__sift = cv2.SIFT_create()
|
||||
self.__bf_matcher = cv2.BFMatcher()
|
||||
|
||||
if load:
|
||||
self.load_db()
|
||||
|
||||
@ -43,6 +46,14 @@ class SIFTDatabase:
|
||||
def size(self, value: Tuple[int, int]):
|
||||
self.__size = value
|
||||
|
||||
@property
|
||||
def sift(self):
|
||||
return self.__sift
|
||||
|
||||
@property
|
||||
def bf_matcher(self):
|
||||
return self.__bf_matcher
|
||||
|
||||
def load_db(self):
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
with conn:
|
||||
@ -75,9 +86,12 @@ class SIFTDatabase:
|
||||
self,
|
||||
__img: Mat,
|
||||
*,
|
||||
sift=cv2.SIFT_create(),
|
||||
bf: cv2.BFMatcher = cv2.BFMatcher(),
|
||||
sift=None,
|
||||
bf=None,
|
||||
) -> Tuple[str, float]:
|
||||
sift = sift or self.sift
|
||||
bf = bf or self.bf_matcher
|
||||
|
||||
img = __img.copy()
|
||||
if self.size is not None:
|
||||
img = cv2.resize(img, self.size)
|
||||
|
Loading…
x
Reference in New Issue
Block a user