cleared the input tensor after use

2022-08-03 09:26:01 +05:30 · 2022-08-03 09:26:01 +05:30 · 730341362e
parent f67219de55
commit 730341362e
1 changed files with 1 additions and 0 deletions
--- a/src/services/machineLearning/ssdMobileNetV2Service.ts
+++ b/src/services/machineLearning/ssdMobileNetV2Service.ts
@ -52,6 +52,7 @@ class SSDMobileNetV2 implements ObjectDetectionService {
            maxNumberBoxes,
            minScore
        );
+        tfImage.dispose();
        return detections;
    }