onnxruntime/js/web/test/e2e/browser-test-wasm-image-tensor-image.js
shalvamist 368d2fc11e
Added E2E test for Image Tensor API (#14406)
### Description
Added E2E test - Currently covering -
URL --> Tensor
ImageData --> Tensor
HTML Image Element --> Tensor
Tensor --> ImageData

---------

Co-authored-by: shalvamist <shalva.mist@microsoft.com>
2023-01-27 08:54:27 -08:00

66 lines
2.2 KiB
JavaScript

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
'use strict';
function getRndColor() {
let r = 255*Math.random()|0,
g = 255*Math.random()|0,
b = 255*Math.random()|0;
return 'rgb(' + r + ',' + g + ',' + b + ')';
}
it('Browser E2E testing - Tensor <--> Image E2E test', async function () {
// Creating Image HTML Image Element
let img = document.createElement('img');
img.crossOrigin = 'Anonymous';
const canvas = document.createElement('canvas');
canvas.height = 200;
canvas.width = 200;
const context = canvas.getContext('2d');
let y, x;
for(y = 0; y < 200; y++) {
for(x = 0; x < 200; x++) {
context.fillStyle = getRndColor();
context.fillRect(x, y, 1, 1);
}
}
img = canvas.toDataURL();
// Image HTML element to tensor API
const inputTensorHTML = await ort.Tensor.fromImage(img);
// Tensor to ImageDAta API
let newImage = inputTensorHTML.toImageData();
// ImageData to tensor API
let inputTensorImageData = await ort.Tensor.fromImage(newImage);
for (let i = 0; i < newImage.height*newImage.width*3; i++) {
if(inputTensorImageData.data[i]!==inputTensorHTML.data[i]){
console.log("Element - " + i + " - " + inputTensorHTML.data[i] + " - " + inputTensorImageData.data[i]);
throw new Error('BUG in ImageData & URL');
}
}
let online = navigator.onLine;
if(online){
// URL element to tensor API
const inputTensorURL = await ort.Tensor.fromImage('https://media.istockphoto.com/id/172859087/photo/square-eggs.jpg?s=2048x2048&w=is&k=20&c=KiBRyyYaoUUSjcJLBh1-qqVu7LW6UQZBopZdva0f5e4=');
// Tensor to ImageDAta API
newImage = inputTensorURL.toImageData();
// ImageData to tensor API
inputTensorImageData = await ort.Tensor.fromImage(newImage);
for (let i = 0; i < newImage.height*newImage.width*3; i++) {
if(inputTensorURL.data[i]!==inputTensorImageData.data[i]){
console.log("Element - " + i + " - " + inputTensorURL.data[i] + " - " + inputTensorImageData.data[i]);
throw new Error('BUG in ImageData & URL');
}
}
}else{
console.log("No internet connection - didn't test Image URL to tensor API");
}
});