Libraryless. Click here for Pure Java version (7473L/50K).
// thread-safe except for analysis output fields sclass ChessPieceRecognizer { MultiMap<S, ChessPieceProfile1> profiles; double magicCombineFactor = 0.0012589254117941495; int version = 1; // for analysis transient Pair<Pair<S, ChessPieceProfile1>, Double> best, runnerUp; void load { long time = sysNow(); profiles = agiBlue_chessPieceProfiles(agiBlue_chessPieceImages_loadedMultiMap(), +version); compact(); done2("Loading " + n2(l(profiles), "piece profile"), time); } void compact { int n = l(profiles); profiles = uniquifyMultiMapValues(profiles); int m = l(profiles); if (m < n) print("Compacted ChessPieceRecognizer " + n + " => " + m); internByteArrayField sideProfile(values(profiles)); } Pair<S, Double> recognize(BufferedImage img) { ChessPieceProfile1 profile = chessOCR_pieceProfileFromRawImage_1(img); new Lowest_withRunnerUp<Pair<S, ChessPieceProfile1>> lowest; for (unpair S piece, ChessPieceProfile1 p : multiMapToPairs(profiles)) { float sideProfileDiff = floatRatio(byteArraysTotalDiff(profile.sideProfile, p.sideProfile), l(p.sideProfile)); float fillGradeDiff = abs(profile.fillGrade-p.fillGrade); double score = magicCombine(sideProfileDiff, fillGradeDiff); lowest.put(pair(piece, p), score); } best = lowest.pair(); runnerUp = lowest.runnerUpPair(); ret !lowest.has() ? null : pair(pairA(lowest!), errorToPercent(lowest.score())); } // for humans double errorToPercent(double d) { ret 100-d; } double magicCombine(double sideProfileScore, double fillGradeScore) { double sum = sideProfileScore+fillGradeScore/magicCombineFactor; //print("magicCombine " + sideProfileScore + "*" + magicCombineFactor + "+" + fillGradeScore + " = " + sum); ret sum; } toString { ret shortClassName(this) + "(v" + version + ", " + nProfiles(l(profiles)) + ")"; } ChessOCR_RecognizedBoard recognizeBoard(BufferedImage board) { new ChessOCR_RecognizedBoard out; LL<BufferedImage> squareImages = bufferedImageMNGrid(board, 8, 8); out.pieces = new L; for (BufferedImage img : concatLists(squareImages)) out.pieces.add(recognize(img)); out.score = doubleAvg(pairsB(out.pieces)); out.fen = chess_makeFEN(listToChunks(8, pairsA(out.pieces))); ret out; } }
download show line numbers debug dex old transpilations
Travelled to 6 computer(s): bhatertpkbcr, mqqgnosmbjvj, pyentgdyhuwx, pzhvpgtvlbxg, tvejysmllsmz, vouqrxazstgt
No comments. add comment
Snippet ID: | #1024689 |
Snippet name: | ChessPieceRecognizer |
Eternal ID of this version: | #1024689/47 |
Text MD5: | 93d99af4deaacecb1663a2d71a8344d5 |
Transpilation MD5: | 435cb478305a5a3887093a8a4c9cae64 |
Author: | stefan |
Category: | javax / image recognition |
Type: | JavaX fragment (include) |
Public (visible to everyone): | Yes |
Archived (hidden from active list): | No |
Created/modified: | 2019-08-28 16:45:35 |
Source code size: | 2433 bytes / 66 lines |
Pitched / IR pitched: | No / No |
Views / Downloads: | 471 / 1003 |
Version history: | 46 change(s) |
Referenced in: | [show references] |