chore: update README

This commit is contained in:
283375 2023-09-07 00:24:39 +08:00
parent 9d8a235bd9
commit 4a9717b7ce
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk

View File

@ -3,32 +3,26 @@
## Example ## Example
```py ```py
import json from arcaea_offline_ocr.device.v2.rois import DeviceV2AutoRois
import pytesseract 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
pytesseract.pytesseract.tesseract_cmd = r'D:/path/to/your/tesseract.exe' knn_model = cv2.ml.KNearest_load(r'/path/to/knn/model')
sift_db = SIFTDatabase(r'/path/to/sift/database.db')
from arcaea_offline_ocr import device, recognize rois = DeviceV2AutoRois(imread_unicode(r'/path/to/your/screenshot.jpg')) # any format that opencv-python supports
ocr = DeviceV2Ocr(knn_model, sift_db)
with open("./assets/devices.json", "r", encoding="utf-8") as file: result = ocr.ocr(rois)
my_device = device.Device.from_json_object(json.loads(file.read())[0]) print(result)
print(recognize.recognize('./assets/screenshots/RMX3370_byd_1.jpg', my_device))
``` ```
![RMX_3370_byd_1.jpg](./assets/screenshots/RMX3370_byd_1.jpg "Screenshot of Arcaea play result: RMX_3370_byd_1.jpg") ```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)
RecognizeResult(pure=38, far=2, lost=7, score=347938, max_recall=21, rating_class=3, title='Kanagawa Cybe')
``` ```
<hr> ## Credits
```py [283375/image-sift-database](https://github.com/283375/image-sift-database)
print(recognize.recognize('./assets/screenshots/RMX3370_ftr_1.jpg', my_device))
```
![RMX_3370_ftr_1.jpg](./assets/screenshots/RMX3370_ftr_1.jpg "Screenshot of Arcaea play result: RMX_3370_ftr_1.jpg")
```
RecognizeResult(pure=1344, far=42, lost=6, score=9807234, max_recall=490, rating_class=2, title='To the Milkv')
```