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;
 | 
			
		||||
 | 
			
		||||
// 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) {
 | 
			
		||||
  EXPECT_EQ(result.embeddings.size(), 1);
 | 
			
		||||
  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 NORM_RECT_STREAM = 'norm_rect';
 | 
			
		||||
const GROUPED_SEGMENTATIONS_STREAM = 'segmented_masks';
 | 
			
		||||
const IMAGEA_SEGMENTER_GRAPH =
 | 
			
		||||
const IMAGE_SEGMENTER_GRAPH =
 | 
			
		||||
    'mediapipe.tasks.vision.image_segmenter.ImageSegmenterGraph';
 | 
			
		||||
 | 
			
		||||
// The OSS JS API does not support the builder pattern.
 | 
			
		||||
| 
						 | 
				
			
			@ -255,7 +255,7 @@ export class ImageSegmenter extends VisionTaskRunner {
 | 
			
		|||
        ImageSegmenterGraphOptionsProto.ext, this.options);
 | 
			
		||||
 | 
			
		||||
    const segmenterNode = new CalculatorGraphConfig.Node();
 | 
			
		||||
    segmenterNode.setCalculator(IMAGEA_SEGMENTER_GRAPH);
 | 
			
		||||
    segmenterNode.setCalculator(IMAGE_SEGMENTER_GRAPH);
 | 
			
		||||
    segmenterNode.addInputStream('IMAGE:' + IMAGE_STREAM);
 | 
			
		||||
    segmenterNode.addInputStream('NORM_RECT:' + NORM_RECT_STREAM);
 | 
			
		||||
    segmenterNode.addOutputStream(
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue
	
	Block a user