Atlas48/grungnet
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated grid size model to be static

Browse Source
This commit is contained in:
Mitchell McCaffrey 2020-10-15 15:54:10 +11:00
parent cc16c8cbf6
commit 396504fd85
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/ml/gridSize/GridSizeModel.js
View File

@ -6,20 +6,21 @@ import config from "./model.json";
import weights from "./group1-shard1of1.bin";
class GridSizeModel extends Model {
model;
// Store model as static to prevent extra network requests
static model;
constructor() {
super(config, { "group1-shard1of1.bin": weights });
}
async predict(imageData) {
if (!this.model) {
this.model = await tf.loadLayersModel(this);
if (!GridSizeModel.model) {
GridSizeModel.model = await tf.loadLayersModel(this);
}
const prediction = tf.tidy(() => {
const image = tf.browser.fromPixels(imageData, 1).toFloat();
const normalized = image.div(tf.scalar(255.0));
const batched = tf.expandDims(normalized);
return this.model.predict(batched);
return GridSizeModel.model.predict(batched);
});
const data = await prediction.data();
return data[0];