mind-ar/src/image-target/detector/kernels/webgl/extremaReduction.js

web augmented reality framework

export const extremaReduction=(args)=>{
    /** @type {import('@tensorflow/tfjs').TensorInfo[]} */
    const {extremasResultT} = args.inputs;
    /** @type {MathBackendWebGL} */
	const backend = args.backend;
    const extremaHeight = extremasResultT.shape[0];
	const extremaWidth = extremasResultT.shape[1];
	const kernel = {
		variableNames: ['extrema'],
		outputShape: [Math.floor(extremaHeight/2), Math.floor(extremaWidth/2)],
		userCode: `
		  void main() {
			ivec2 coords = getOutputCoords();
			int y = coords[0] * 2;
			int x = coords[1] * 2;
  
			float location = 0.0;
			float values = getExtrema(y, x);
  
			if (getExtrema(y+1, x) != 0.0) {
			  location = 1.0;
		  values = getExtrema(y+1, x);
			}
			else if (getExtrema(y, x+1) != 0.0) {
			  location = 2.0;
		  values = getExtrema(y, x+1);
			}
			else if (getExtrema(y+1, x+1) != 0.0) {
			  location = 3.0;
		  values = getExtrema(y+1, x+1);
			}
  
			if (values < 0.0) {
			  setOutput(location * -1000.0 + values);
			} else {
			  setOutput(location * 1000.0 + values);
			}
		  }
		`
	}

	return backend.runWebGLProgram(kernel,[extremasResultT],extremasResultT.dtype);
}

export const extremaReductionConfig = {//: KernelConfig
    kernelName: "ExtremaReduction",
    backendName: 'webgl',
    kernelFunc: extremaReduction,// as {} as KernelFunc,
};