beenull/ente
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

remove possible memory leaks

Browse source
This commit is contained in:
Abhinav 2022-08-03 10:38:53 +05:30
parent b866745fe7
commit 583f35a32f
2 changed files with 2 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/services/machineLearning/imageSceneService.ts
View file

@ -33,7 +33,7 @@ class ImageScene implements SceneDetectionService {
const model = await tfjsConverter.loadGraphModel( const model = await tfjsConverter.loadGraphModel(
'/models/imagescene/model.json' '/models/imagescene/model.json'
); );
console.log('loaded ImageScene model', model, tf.getBackend()); console.log('loaded ImageScene model', tf.getBackend());
this.model = model; this.model = model;
// warmup the model // warmup the model

6
src/services/machineLearning/ssdMobileNetV2Service.ts
View file

@ -25,11 +25,7 @@ class SSDMobileNetV2 implements ObjectDetectionService {
base: 'mobilenet_v2', base: 'mobilenet_v2',
modelUrl: '/models/ssdmobilenet/model.json', modelUrl: '/models/ssdmobilenet/model.json',
}); });
console.log( console.log('loaded ssdMobileNetV2Model', tf.getBackend());
'loaded ssdMobileNetV2Model',
this.ssdMobileNetV2Model,
tf.getBackend()
);
} }
private async getSSDMobileNetV2Model() { private async getSSDMobileNetV2Model() {