pass modelURL while loading SSDMobileNet to load the custom model
This commit is contained in:
parent
f2df2d9ad1
commit
9fcffec0e6
|
@ -29,6 +29,7 @@ class SSDMobileNetV2 implements ObjectDetectionService {
|
||||||
this.ssdMobileNetV2Model = await SSDMobileNet.load({
|
this.ssdMobileNetV2Model = await SSDMobileNet.load({
|
||||||
base: 'mobilenet_v2',
|
base: 'mobilenet_v2',
|
||||||
dataset: 'open_images',
|
dataset: 'open_images',
|
||||||
|
modelUrl: '/models/open-images-ssd-mobilenet-v2/model.json',
|
||||||
});
|
});
|
||||||
console.log(
|
console.log(
|
||||||
'loaded ssdMobileNetV2Model',
|
'loaded ssdMobileNetV2Model',
|
||||||
|
|
Loading…
Reference in a new issue