mirror of
https://github.com/opencv/opencv.git
synced 2024-12-16 02:19:12 +08:00
99672a2691
modules/js/perf/perf_helpfunc.js and target tests, e.g. perf_gaussianBlur.js contained "const isNodeJs", leading to re-definition when using associated *.html files.
180 lines
5.0 KiB
JavaScript
180 lines
5.0 KiB
JavaScript
var isNodeJs = (typeof window) === 'undefined'? true : false;
|
||
|
||
if (isNodeJs) {
|
||
var Benchmark = require('benchmark');
|
||
var cv = require('../../opencv');
|
||
} else {
|
||
var paramsElement = document.getElementById('params');
|
||
var runButton = document.getElementById('runButton');
|
||
var logElement = document.getElementById('log');
|
||
}
|
||
|
||
function perf() {
|
||
|
||
console.log('opencv.js loaded');
|
||
if (isNodeJs) {
|
||
global.cv = cv;
|
||
} else {
|
||
runButton.removeAttribute('disabled');
|
||
runButton.setAttribute('class', 'btn btn-primary');
|
||
runButton.innerHTML = 'Run';
|
||
}
|
||
let totalCaseNum, currentCaseId;
|
||
|
||
|
||
function addCountNonZeroCase(suite) {
|
||
suite.add('countNonZero', function() {
|
||
cv.countNonZero(mat);
|
||
}, {
|
||
'setup': function() {
|
||
let size = this.params.size;
|
||
let mat = cv.Mat.eye(size[0], size[1], cv.CV_64F);
|
||
}, 'teardown': function() {
|
||
mat.delete();
|
||
}
|
||
});
|
||
}
|
||
|
||
function addMatDotCase(suite) {
|
||
suite.add('Mat::dot', function() {
|
||
mat.dot(matT);
|
||
}, {
|
||
'setup': function() {
|
||
let size = this.params.size;
|
||
let mat = cv.Mat.ones(size[0], size[1], cv.CV_64FC1);
|
||
let matT = mat.t();
|
||
}, 'teardown': function() {
|
||
mat.delete();
|
||
matT.delete();
|
||
}
|
||
});
|
||
}
|
||
|
||
function addSplitCase(suite) {
|
||
suite.add('Split', function() {
|
||
cv.split(mat, planes);
|
||
}, {
|
||
'setup': function() {
|
||
let size = this.params.size;
|
||
let mat = cv.Mat.ones(size[0], size[1], cv.CV_64FC3);
|
||
let planes = new cv.MatVector();
|
||
}, 'teardown': function() {
|
||
mat.delete();
|
||
planes.delete();
|
||
}
|
||
});
|
||
}
|
||
|
||
function addMergeCase(suite) {
|
||
suite.add('Merge', function() {
|
||
cv.merge(planes, mat);
|
||
}, {
|
||
'setup': function() {
|
||
let size = this.params.size;
|
||
let mat = new cv.Mat();
|
||
let mat1 = cv.Mat.ones(size[0], size[1], cv.CV_64FC3);
|
||
let planes = new cv.MatVector();
|
||
cv.split(mat1, planes);
|
||
}, 'teardown': function() {
|
||
mat.delete();
|
||
mat1.delete();
|
||
planes.delete();
|
||
}
|
||
});
|
||
}
|
||
|
||
function setInitParams(suite, sizeArray) {
|
||
for( let i =0; i < suite.length; i++) {
|
||
suite[i].params = {
|
||
size: sizeArray
|
||
};
|
||
}
|
||
}
|
||
|
||
function log(message) {
|
||
console.log(message);
|
||
if (!isNodeJs) {
|
||
logElement.innerHTML += `\n${'\t' + message}`;
|
||
}
|
||
}
|
||
|
||
function setBenchmarkSuite(suite) {
|
||
suite
|
||
// add listeners
|
||
.on('cycle', function(event) {
|
||
++currentCaseId;
|
||
let size = event.target.params.size;
|
||
log(`=== ${event.target.name} ${currentCaseId} ===`);
|
||
log(`params: (${parseInt(size[0])}x${parseInt(size[1])})`);
|
||
log('elapsed time:' +String(event.target.times.elapsed*1000)+' ms');
|
||
log('mean time:' +String(event.target.stats.mean*1000)+' ms');
|
||
log('stddev time:' +String(event.target.stats.deviation*1000)+' ms');
|
||
log(String(event.target));
|
||
})
|
||
.on('error', function(event) { log(`test case ${event.target.name} failed`); })
|
||
.on('complete', function(event) {
|
||
log(`\n ###################################`)
|
||
log(`Finished testing ${event.currentTarget.length} cases \n`);
|
||
if (!isNodeJs) {
|
||
runButton.removeAttribute('disabled');
|
||
runButton.setAttribute('class', 'btn btn-primary');
|
||
runButton.innerHTML = 'Run';
|
||
}
|
||
});
|
||
}
|
||
|
||
function genBenchmarkCase(paramsContent) {
|
||
let suite = new Benchmark.Suite;
|
||
var sizeArray;
|
||
totalCaseNum = 4;
|
||
currentCaseId = 0;
|
||
if (/\([0-9]+x[0-9]+\)/g.test(paramsContent.toString())) {
|
||
let params = paramsContent.toString().match(/\([0-9]+x[0-9]+\)/g)[0];
|
||
let sizeStrs = (params.match(/[0-9]+/g) || []).slice(0, 2).toString().split(",");
|
||
sizeArray = sizeStrs.map(Number);
|
||
} else {
|
||
log("no getting invalid params, run all the cases with Mat of shape (1000 x 1000)");
|
||
sizeArray = [1000, 1000];
|
||
}
|
||
addCountNonZeroCase(suite);
|
||
addMatDotCase(suite);
|
||
addSplitCase(suite);
|
||
addMergeCase(suite);
|
||
setInitParams(suite, sizeArray)
|
||
setBenchmarkSuite(suite);
|
||
log(`Running ${totalCaseNum} tests from 64-bit intrinsics`);
|
||
suite.run({ 'async': true }); // run the benchmark
|
||
}
|
||
|
||
|
||
// set test filter params
|
||
if (isNodeJs) {
|
||
const args = process.argv.slice(2);
|
||
let paramsContent = '';
|
||
if (/--test_param_filter=\([0-9]+x[0-9]+,[\ ]*\w+\)/g.test(args.toString())) {
|
||
paramsContent = args.toString().match(/\([0-9]+x[0-9]+,[\ ]*\w+\)/g)[0];
|
||
}
|
||
genBenchmarkCase(paramsContent);
|
||
} else {
|
||
runButton.onclick = function() {
|
||
let paramsContent = paramsElement.value;
|
||
genBenchmarkCase(paramsContent);
|
||
if (totalCaseNum !== 0) {
|
||
runButton.setAttribute("disabled", "disabled");
|
||
runButton.setAttribute('class', 'btn btn-primary disabled');
|
||
runButton.innerHTML = "Running";
|
||
}
|
||
}
|
||
}
|
||
};
|
||
|
||
async function main() {
|
||
if (cv instanceof Promise) {
|
||
cv = await cv;
|
||
perf();
|
||
} else {
|
||
cv.onRuntimeInitialized = perf;
|
||
}
|
||
}
|
||
|
||
main(); |