Skip to content

Commit 39ce6d6

Browse files
fix(image-segmenter): Add "selfie_segmenter_landscape" model and fix "selfie_segmenter" model URL
fix(image-segmenter): Add "selfie_segmenter_landscape" model and fix "selfie_segmenter" model URL
2 parents 6209506 + 1c80bbd commit 39ce6d6

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

src/tasks/image-segmenter.ts

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,9 @@ const standardModels: Record<string, string> = {
5050
hair_segmenter:
5151
'https://storage.googleapis.com/mediapipe-models/image_segmenter/hair_segmenter/float32/1/hair_segmenter.tflite',
5252
selfie_segmenter:
53-
'https://storage.googleapis.com/mediapipe-models/image_segmenter/selfie_segmenter/float32/latest/selfie_segmenter.tflite',
53+
'https://storage.googleapis.com/mediapipe-models/image_segmenter/selfie_segmenter/float16/latest/selfie_segmenter.tflite',
54+
selfie_segmenter_landscape:
55+
'https://storage.googleapis.com/mediapipe-models/image_segmenter/selfie_segmenter_landscape/float16/latest/selfie_segmenter_landscape.tflite',
5456
selfie_multiclass:
5557
'https://storage.googleapis.com/mediapipe-models/image_segmenter/selfie_multiclass_256x256/float32/latest/selfie_multiclass_256x256.tflite',
5658
};
@@ -84,6 +86,7 @@ class ImageSegmenterTask extends BaseVisionTask {
8486
{ label: 'DeepLab V3', value: 'deeplab_v3', isDefault: true },
8587
{ label: 'Hair Segmenter', value: 'hair_segmenter' },
8688
{ label: 'Selfie Segmenter', value: 'selfie_segmenter' },
89+
{ label: 'Selfie Segmenter Landscape', value: 'selfie_segmenter_landscape' },
8790
{ label: 'Selfie Multi-class', value: 'selfie_multiclass' },
8891
]);
8992
}

0 commit comments

Comments
 (0)