Skip to content

Instantly share code, notes, and snippets.

@zaidalyafeai
Last active March 30, 2019 20:16
Show Gist options
  • Save zaidalyafeai/e03f072a17d6069e6e274d98429b815d to your computer and use it in GitHub Desktop.
Save zaidalyafeai/e03f072a17d6069e6e274d98429b815d to your computer and use it in GitHub Desktop.
<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs/dist/tf.min.js"> </script>
<script src="https://cdn.jsdelivr.net/npm/ua-parser-js@0/dist/ua-parser.min.js"></script>
<img id="img" src="https://i.imgur.com/gQLIJkN.png" crossorigin="" style = 'display:none;' />
<table>
<td>Browser</td>
<td id = 'browserName'></td>
<td id = 'browserVersion'></td>
</tr>
<tr>
<td>OS</td>
<td id = 'osName'></td>
<td id = 'osVersion'></td>
</tr>
<tr>
<td>Time</td>
<td id = 'time'></td>
<td >ms</td>
</tr>
</table>
<script>
var parser = new UAParser();
var output = parser.getResult();
document.getElementById('browserVersion').innerHTML = output.browser.version;
document.getElementById('browserName').innerHTML = output.browser.name;
document.getElementById('osName').innerHTML = output.os.name;
document.getElementById('osVersion').innerHTML = output.os.version;
const path = 'https://storage.googleapis.com/tfjs-models/tfjs/mobilenet_v1_1.0_224/model.json';
tf.loadLayersModel(path).then(model => {
//warm up
model.predict(tf.zeros([1, 224, 224, 3]));
//preprocess
image = document.getElementById('img');
const tensor = tf.browser.fromPixels(image);
const batch = tf.expandDims(tensor);
const input = tf.image.resizeBilinear (batch, [224, 224]);
var image;
var time = 0;
const num_iter = 200;
for (i= 1 ; i <= num_iter; i++){
const t0 = tf.util.now ();
const output = model.predict(input);
output.dataSync();
const t1 = tf.util.now ();
time += (t1-t0);
}
document.getElementById('time').innerHTML = (time/num_iter);
});
</script>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment