mirror of
https://github.com/TheFunny/ArisuAutoSweeper
synced 2026-10-23 04:52:06 +00:00
perf: optimize image cropping by avoiding unnecessary copies
This commit is contained in:
+2
-2
@@ -119,7 +119,7 @@ class Ocr:
|
|||||||
# pre process
|
# pre process
|
||||||
start_time = time.time()
|
start_time = time.time()
|
||||||
if not direct_ocr:
|
if not direct_ocr:
|
||||||
image = crop(image, self.button.area)
|
image = crop(image, self.button.area, copy=False)
|
||||||
image = self.pre_process(image)
|
image = self.pre_process(image)
|
||||||
# ocr
|
# ocr
|
||||||
result, _ = self.model.ocr_single_line(image)
|
result, _ = self.model.ocr_single_line(image)
|
||||||
@@ -162,7 +162,7 @@ class Ocr:
|
|||||||
# pre process
|
# pre process
|
||||||
start_time = time.time()
|
start_time = time.time()
|
||||||
if not direct_ocr:
|
if not direct_ocr:
|
||||||
image = crop(image, self.button.area)
|
image = crop(image, self.button.area, copy=False)
|
||||||
image = self.pre_process(image)
|
image = self.pre_process(image)
|
||||||
# ocr
|
# ocr
|
||||||
results: list[BoxedResult] = self.model.detect_and_ocr(image)
|
results: list[BoxedResult] = self.model.detect_and_ocr(image)
|
||||||
|
|||||||
Reference in New Issue
Block a user