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

chore: update README

Browse Source
This commit is contained in:
283375 2023-11-01 14:01:53 +08:00
parent 58d5022952
commit 288ccddfa1
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
1 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
README.md
View File

@ -3,26 +3,32 @@
## Example
```py
from arcaea_offline_ocr.device.v2.rois import DeviceV2AutoRois
from arcaea_offline_ocr.device.v2.ocr import DeviceV2Ocr
from arcaea_offline_ocr.sift_db import SIFTDatabase
from arcaea_offline_ocr.utils import imread_unicode
import cv2
from arcaea_offline_ocr.device.ocr import DeviceOcr
from arcaea_offline_ocr.device.rois.definition import DeviceRoisAutoT2
from arcaea_offline_ocr.device.rois.extractor import DeviceRoisExtractor
from arcaea_offline_ocr.device.rois.masker import DeviceRoisMaskerAutoT2
from arcaea_offline_ocr.phash_db import ImagePhashDatabase
knn_model = cv2.ml.KNearest_load(r'/path/to/knn/model')
sift_db = SIFTDatabase(r'/path/to/sift/database.db')
img_path = "/path/to/opencv/supported/image/formats.jpg"
img = cv2.imread(img_path, cv2.IMREAD_COLOR)
rois = DeviceV2AutoRois(imread_unicode(r'/path/to/your/screenshot.jpg')) # any format that opencv-python supports
ocr = DeviceV2Ocr(knn_model, sift_db)
result = ocr.ocr(rois)
print(result)
rois = DeviceRoisAutoT2(img.shape[1], img.shape[0])
extractor = DeviceRoisExtractor(img, rois)
masker = DeviceRoisMaskerAutoT2()
knn_model = cv2.ml.KNearest.load("/path/to/trained/knn/model.dat")
phash_db = ImagePhashDatabase("/path/to/image/phash/database.db")
ocr = DeviceOcr(extractor, masker, knn_model, phash_db)
print(ocr.ocr())
```
```sh
$ python example.py
DeviceOcrResult(rating_class=2, pure=1371, far=62, lost=34, score=9558078, max_recall=330, song_id='abstrusedilemma', title=None, clear_type=None)
DeviceOcrResult(rating_class=2, pure=1135, far=11, lost=0, score=9953016, max_recall=1146, song_id='ringedgenesis', song_id_possibility=0.953125, clear_status=2, partner_id='45', partner_id_possibility=0.8046875)
```
## Credits
[283375/image-sift-database](https://github.com/283375/image-sift-database)
[283375/image-phash-database](https://github.com/283375/image-phash-database)