Remove unnecessary "disposes" of tensors

hugozanini · Jun 3, 2020 · 0f8232e · 0f8232e
1 parent d032372
commit 0f8232e
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/src/index.js b/src/index.js
@@ -63,7 +63,6 @@ class App extends React.Component {
       tf.engine().startScope();
       const predictions = model.predict(this.process_input(video));
       this.renderPredictions(predictions);
-      predictions.dispose();
       requestAnimationFrame(() => {
         this.detectFrame(video, model);
       });
@@ -77,12 +76,6 @@ class App extends React.Component {
     const std = tf.tensor3d([0.229, 0.224, 0.225], [1,1,3]);
     const normalised = img.div(scale).sub(mean).div(std);
     const batched = normalised.transpose([2,0,1]).expandDims();
-    img.dispose();
-    scale.dispose();
-    mean.dispose();
-    std.dispose();
-    normalised.dispose();
-    batched.dispose();
     return batched;
   };
     renderPredictions = async (predictions) => {