优化后端
This commit is contained in:
Generated
Vendored
+21
@@ -0,0 +1,21 @@
|
||||
#!/usr/bin/env node
|
||||
|
||||
'use strict';
|
||||
|
||||
const path = require('path');
|
||||
const fs = require('fs');
|
||||
const { createWorker } = require('../..');
|
||||
|
||||
const [,, imagePath] = process.argv;
|
||||
const image = path.resolve(__dirname, (imagePath || '../../tests/assets/images/cosmic.png'));
|
||||
|
||||
console.log(`Recognizing ${image}`);
|
||||
|
||||
(async () => {
|
||||
const worker = await createWorker();
|
||||
const { data: { text, pdf } } = await worker.recognize(image, { pdfTitle: 'Example PDF' }, { pdf: true });
|
||||
console.log(text);
|
||||
fs.writeFileSync('tesseract-ocr-result.pdf', Buffer.from(pdf));
|
||||
console.log('Generate PDF: tesseract-ocr-result.pdf');
|
||||
await worker.terminate();
|
||||
})();
|
||||
Generated
Vendored
+37
@@ -0,0 +1,37 @@
|
||||
#!/usr/bin/env node
|
||||
|
||||
'use strict';
|
||||
|
||||
const path = require('path');
|
||||
const fs = require('fs');
|
||||
const { createWorker } = require('../..');
|
||||
|
||||
const [,, imagePath] = process.argv;
|
||||
const image = path.resolve(__dirname, (imagePath || '../data/meditations.jpg'));
|
||||
|
||||
console.log(`Recognizing ${image}`);
|
||||
|
||||
// Tesseract.js returns images (imageColor, imageGrey, imageBinary) as strings
|
||||
// to be used as source tags.
|
||||
// This function converts to Uint8Array data for saving to disk.
|
||||
const convertImage = (imageSrc) => {
|
||||
const data = atob(imageSrc.split(',')[1])
|
||||
.split('')
|
||||
.map((c) => c.charCodeAt(0));
|
||||
|
||||
return new Uint8Array(data);
|
||||
};
|
||||
|
||||
(async () => {
|
||||
const worker = await createWorker();
|
||||
const { data: { imageColor, imageGrey, imageBinary } } = await worker.recognize(image,
|
||||
{ rotateAuto: true }, { imageColor: true, imageGrey: true, imageBinary: true });
|
||||
|
||||
console.log('Saving intermediate images: imageColor.png, imageGrey.png, imageBinary.png');
|
||||
|
||||
fs.writeFileSync('imageColor.png', convertImage(imageColor));
|
||||
fs.writeFileSync('imageGrey.png', convertImage(imageGrey));
|
||||
fs.writeFileSync('imageBinary.png', convertImage(imageBinary));
|
||||
|
||||
await worker.terminate();
|
||||
})();
|
||||
+20
@@ -0,0 +1,20 @@
|
||||
#!/usr/bin/env node
|
||||
|
||||
'use strict';
|
||||
|
||||
const path = require('path');
|
||||
const { createWorker } = require('../..');
|
||||
|
||||
const [,, imagePath] = process.argv;
|
||||
const image = path.resolve(__dirname, (imagePath || '../../tests/assets/images/cosmic.png'));
|
||||
|
||||
console.log(`Recognizing ${image}`);
|
||||
|
||||
(async () => {
|
||||
const worker = await createWorker('eng', 1, {
|
||||
logger: (m) => console.log(m),
|
||||
});
|
||||
const { data: { text } } = await worker.recognize(image);
|
||||
console.log(text);
|
||||
await worker.terminate();
|
||||
})();
|
||||
+40
@@ -0,0 +1,40 @@
|
||||
'use strict';
|
||||
|
||||
const path = require('path');
|
||||
const { createWorker, createScheduler } = require('../..');
|
||||
|
||||
const [,, imagePath] = process.argv;
|
||||
|
||||
// Note: This example recognizes the same image 4 times in parallel
|
||||
// to show how schedulers can be used to speed up bulk jobs.
|
||||
// In actual use you would (obviously) not want to run multiple identical jobs.
|
||||
|
||||
const image = path.resolve(__dirname, (imagePath || '../../tests/assets/images/cosmic.png'));
|
||||
const imageArr = [image, image, image, image];
|
||||
|
||||
const scheduler = createScheduler();
|
||||
|
||||
// Creates worker and adds to scheduler
|
||||
const workerGen = async () => {
|
||||
const worker = await createWorker('eng', 1, { cachePath: '.' });
|
||||
scheduler.addWorker(worker);
|
||||
};
|
||||
|
||||
const workerN = 4;
|
||||
(async () => {
|
||||
const resArr = Array(workerN);
|
||||
for (let i = 0; i < workerN; i++) {
|
||||
resArr[i] = workerGen();
|
||||
}
|
||||
await Promise.all(resArr);
|
||||
|
||||
const resArr2 = Array(imageArr.length);
|
||||
|
||||
for (let i = 0; i < imageArr.length; i++) {
|
||||
resArr2[i] = scheduler.addJob('recognize', image).then((x) => console.log(x.data.text));
|
||||
}
|
||||
|
||||
await Promise.all(resArr2);
|
||||
|
||||
await scheduler.terminate(); // It also terminates all workers.
|
||||
})();
|
||||
Reference in New Issue
Block a user