Merge branch 'google:master' into face-landmarker-python
This commit is contained in:
commit
2af660321d
|
@ -52,7 +52,7 @@ constexpr char kMobileNetV3Embedder[] =
|
||||||
constexpr double kSimilarityTolerancy = 1e-6;
|
constexpr double kSimilarityTolerancy = 1e-6;
|
||||||
|
|
||||||
// Utility function to check the sizes, head_index and head_names of a result
|
// Utility function to check the sizes, head_index and head_names of a result
|
||||||
// procduced by kMobileNetV3Embedder.
|
// produced by kMobileNetV3Embedder.
|
||||||
void CheckMobileNetV3Result(const ImageEmbedderResult& result, bool quantized) {
|
void CheckMobileNetV3Result(const ImageEmbedderResult& result, bool quantized) {
|
||||||
EXPECT_EQ(result.embeddings.size(), 1);
|
EXPECT_EQ(result.embeddings.size(), 1);
|
||||||
EXPECT_EQ(result.embeddings[0].head_index, 0);
|
EXPECT_EQ(result.embeddings[0].head_index, 0);
|
||||||
|
|
|
@ -35,7 +35,7 @@ export {ImageSource}; // Used in the public API
|
||||||
const IMAGE_STREAM = 'image_in';
|
const IMAGE_STREAM = 'image_in';
|
||||||
const NORM_RECT_STREAM = 'norm_rect';
|
const NORM_RECT_STREAM = 'norm_rect';
|
||||||
const GROUPED_SEGMENTATIONS_STREAM = 'segmented_masks';
|
const GROUPED_SEGMENTATIONS_STREAM = 'segmented_masks';
|
||||||
const IMAGEA_SEGMENTER_GRAPH =
|
const IMAGE_SEGMENTER_GRAPH =
|
||||||
'mediapipe.tasks.vision.image_segmenter.ImageSegmenterGraph';
|
'mediapipe.tasks.vision.image_segmenter.ImageSegmenterGraph';
|
||||||
|
|
||||||
// The OSS JS API does not support the builder pattern.
|
// The OSS JS API does not support the builder pattern.
|
||||||
|
@ -255,7 +255,7 @@ export class ImageSegmenter extends VisionTaskRunner {
|
||||||
ImageSegmenterGraphOptionsProto.ext, this.options);
|
ImageSegmenterGraphOptionsProto.ext, this.options);
|
||||||
|
|
||||||
const segmenterNode = new CalculatorGraphConfig.Node();
|
const segmenterNode = new CalculatorGraphConfig.Node();
|
||||||
segmenterNode.setCalculator(IMAGEA_SEGMENTER_GRAPH);
|
segmenterNode.setCalculator(IMAGE_SEGMENTER_GRAPH);
|
||||||
segmenterNode.addInputStream('IMAGE:' + IMAGE_STREAM);
|
segmenterNode.addInputStream('IMAGE:' + IMAGE_STREAM);
|
||||||
segmenterNode.addInputStream('NORM_RECT:' + NORM_RECT_STREAM);
|
segmenterNode.addInputStream('NORM_RECT:' + NORM_RECT_STREAM);
|
||||||
segmenterNode.addOutputStream(
|
segmenterNode.addOutputStream(
|
||||||
|
|
Loading…
Reference in New Issue
Block a user