pass modelURL while loading SSDMobileNet to load the custom model

2022-02-22 08:17:50 +05:30 · 2022-02-22 08:17:50 +05:30 · 9fcffec0e6
parent f2df2d9ad1
commit 9fcffec0e6
1 changed files with 1 additions and 0 deletions
--- a/src/services/machineLearning/ssdMobileNetV2Service.ts
+++ b/src/services/machineLearning/ssdMobileNetV2Service.ts
@ -29,6 +29,7 @@ class SSDMobileNetV2 implements ObjectDetectionService {
        this.ssdMobileNetV2Model = await SSDMobileNet.load({
            base: 'mobilenet_v2',
            dataset: 'open_images',
+            modelUrl: '/models/open-images-ssd-mobilenet-v2/model.json',
        });
        console.log(
            'loaded ssdMobileNetV2Model',