working on crop
This commit is contained in:
parent
93e2234c4e
commit
f74e3e1c85
12 changed files with 1135 additions and 220 deletions
|
|
@ -1,7 +1,7 @@
|
|||
import { mkdir } from 'fs/promises';
|
||||
import { join } from 'path';
|
||||
import { logger } from '../util/logger.js';
|
||||
import { OcrDaemon, type OcrResponse, type OcrEngine, type OcrPreprocess, type DiffOcrParams, type DiffOcrResponse, type TemplateMatchResult } from './OcrDaemon.js';
|
||||
import { OcrDaemon, type OcrResponse, type OcrEngine, type OcrPreprocess, type DiffOcrParams, type DiffCropParams, type OcrParams, type DiffOcrResponse, type TemplateMatchResult } from './OcrDaemon.js';
|
||||
import { GridReader, type GridLayout, type CellCoord } from './GridReader.js';
|
||||
import type { Region } from '../types.js';
|
||||
|
||||
|
|
@ -23,11 +23,28 @@ export class ScreenReader {
|
|||
settings: OcrSettings = {
|
||||
engine: 'easyocr',
|
||||
screenPreprocess: 'none',
|
||||
tooltipPreprocess: 'bgsub',
|
||||
tooltipParams: {},
|
||||
tooltipPreprocess: 'tophat',
|
||||
tooltipParams: {
|
||||
crop: { diffThresh: 10 },
|
||||
ocr: { kernelSize: 21 },
|
||||
},
|
||||
saveDebugImages: true,
|
||||
};
|
||||
|
||||
/**
|
||||
* Eagerly spawn the OCR daemon and warm up the EasyOCR model.
|
||||
* Fire-and-forget a small OCR request so the Python model loads in the background.
|
||||
*/
|
||||
async warmup(): Promise<void> {
|
||||
await this.daemon.warmup();
|
||||
// Fire a small EasyOCR request to trigger Python model load
|
||||
// Use a tiny 1×1 region to minimize work, we only care about loading the model
|
||||
const { engine } = this.settings;
|
||||
if (engine !== 'tesseract') {
|
||||
await this.daemon.ocr({ x: 0, y: 0, width: 100, height: 100 }, engine);
|
||||
}
|
||||
}
|
||||
|
||||
// ── Screenshot capture ──────────────────────────────────────────────
|
||||
|
||||
async captureScreen(): Promise<Buffer> {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue