Project import generated by Copybara.

GitOrigin-RevId: 08c2016a4df5aef571b464a4d4491f38c6b2af10

mediapipe/examples/android/src/java/com/google/mediapipe/apps/selfiesegmentationgpu/BUILD

@@ -0,0 +1,60 @@
+# Copyright 2021 The MediaPipe Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+licenses(["notice"])
+
+package(default_visibility = ["//visibility:private"])
+
+cc_binary(
+    name = "libmediapipe_jni.so",
+    linkshared = 1,
+    linkstatic = 1,
+    deps = [
+        "//mediapipe/graphs/selfie_segmentation:selfie_segmentation_gpu_deps",
+        "//mediapipe/java/com/google/mediapipe/framework/jni:mediapipe_framework_jni",
+    ],
+)
+
+cc_library(
+    name = "mediapipe_jni_lib",
+    srcs = [":libmediapipe_jni.so"],
+    alwayslink = 1,
+)
+
+android_binary(
+    name = "selfiesegmentationgpu",
+    srcs = glob(["*.java"]),
+    assets = [
+        "//mediapipe/graphs/selfie_segmentation:selfie_segmentation_gpu.binarypb",
+        "//mediapipe/modules/selfie_segmentation:selfie_segmentation.tflite",
+    ],
+    assets_dir = "",
+    manifest = "//mediapipe/examples/android/src/java/com/google/mediapipe/apps/basic:AndroidManifest.xml",
+    manifest_values = {
+        "applicationId": "com.google.mediapipe.apps.selfiesegmentationgpu",
+        "appName": "Selfie Segmentation",
+        "mainActivity": "com.google.mediapipe.apps.basic.MainActivity",
+        "cameraFacingFront": "True",
+        "binaryGraphName": "selfie_segmentation_gpu.binarypb",
+        "inputVideoStreamName": "input_video",
+        "outputVideoStreamName": "output_video",
+        "flipFramesVertically": "True",
+        "converterNumBuffers": "2",
+    },
+    multidex = "native",
+    deps = [
+        ":mediapipe_jni_lib",
+        "//mediapipe/examples/android/src/java/com/google/mediapipe/apps/basic:basic_lib",
+    ],
+)

mediapipe/examples/desktop/autoflip/calculators/face_box_adjuster_calculator.proto

@@ -49,7 +49,23 @@ message FaceBoxAdjusterCalculatorOptions {
   optional float ipd_face_box_height_ratio = 7 [default = 0.3131];
 
   // The max look up angle before considering the eye distance unstable.
-  optional float max_head_tilt_angle_deg = 8 [default = 12.0];
+  optional float max_head_tilt_angle_deg = 8 [default = 5.0];
+  // The min look up angle (i.e. looking down) before considering the eye
+  // distance unstable.
+  optional float min_head_tilt_angle_deg = 10 [default = -18.0];
+  // The max look right angle before considering the eye distance unstable.
+  optional float max_head_pan_angle_deg = 11 [default = 25.0];
+  // The min look right angle (i.e. looking left) before considering the eye
+  // distance unstable.
+  optional float min_head_pan_angle_deg = 12 [default = -25.0];
+
+  // Update rate for motion history, valid values [0.0, 1.0].
+  optional float motion_history_alpha = 13 [default = 0.5];
+
+  // Max value of head motion (max of current or history) to be considered still
+  // stable.
+  optional float head_motion_threshold = 14 [default = 10.0];
+
   // The max amount of time to use an old eye distance when the face look angle
   // is unstable.
   optional int32 max_facesize_history_us = 9 [default = 8000000];

mediapipe/examples/desktop/autoflip/subgraph/face_detection_subgraph.pbtxt

@@ -14,8 +14,8 @@ node: {
   output_stream: "LETTERBOX_PADDING:letterbox_padding"
   options: {
     [mediapipe.ImageTransformationCalculatorOptions.ext] {
-      output_width: 256
-      output_height: 256
+      output_width: 192
+      output_height: 192
       scale_mode: FIT
     }
   }
@@ -50,19 +50,17 @@ node {
   output_side_packet: "anchors"
   options: {
     [mediapipe.SsdAnchorsCalculatorOptions.ext] {
-      num_layers: 4
-      min_scale: 0.15625
+      num_layers: 1
+      min_scale: 0.1484375
       max_scale: 0.75
-      input_size_height: 256
-      input_size_width: 256
+      input_size_height: 192
+      input_size_width: 192
       anchor_offset_x: 0.5
       anchor_offset_y: 0.5
-      strides: 16
-      strides: 32
-      strides: 32
-      strides: 32
+      strides: 4
       aspect_ratios: 1.0
       fixed_anchor_size: true
+      interpolated_scale_aspect_ratio: 0.0
     }
   }
 }
@@ -78,7 +76,7 @@ node {
   options: {
     [mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.ext] {
       num_classes: 1
-      num_boxes: 896
+      num_boxes: 2304
       num_coords: 16
       box_coord_offset: 0
       keypoint_coord_offset: 4
@@ -87,11 +85,11 @@ node {
       sigmoid_score: true
       score_clipping_thresh: 100.0
       reverse_output_order: true
-      x_scale: 256.0
-      y_scale: 256.0
-      h_scale: 256.0
-      w_scale: 256.0
-      min_score_thresh: 0.65
+      x_scale: 192.0
+      y_scale: 192.0
+      h_scale: 192.0
+      w_scale: 192.0
+      min_score_thresh: 0.6
     }
   }
 }

mediapipe/examples/desktop/selfie_segmentation/BUILD

@@ -0,0 +1,34 @@
+# Copyright 2021 The MediaPipe Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+licenses(["notice"])
+
+package(default_visibility = ["//mediapipe/examples:__subpackages__"])
+
+cc_binary(
+    name = "selfie_segmentation_cpu",
+    deps = [
+        "//mediapipe/examples/desktop:demo_run_graph_main",
+        "//mediapipe/graphs/selfie_segmentation:selfie_segmentation_cpu_deps",
+    ],
+)
+
+# Linux only
+cc_binary(
+    name = "selfie_segmentation_gpu",
+    deps = [
+        "//mediapipe/examples/desktop:demo_run_graph_main_gpu",
+        "//mediapipe/graphs/selfie_segmentation:selfie_segmentation_gpu_deps",
+    ],
+)

mediapipe/examples/ios/selfiesegmentationgpu/BUILD

@@ -0,0 +1,69 @@
+# Copyright 2021 The MediaPipe Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+load(
+    "@build_bazel_rules_apple//apple:ios.bzl",
+    "ios_application",
+)
+load(
+    "//mediapipe/examples/ios:bundle_id.bzl",
+    "BUNDLE_ID_PREFIX",
+    "example_provisioning",
+)
+
+licenses(["notice"])
+
+MIN_IOS_VERSION = "10.0"
+
+alias(
+    name = "selfiesegmentationgpu",
+    actual = "SelfieSegmentationGpuApp",
+)
+
+ios_application(
+    name = "SelfieSegmentationGpuApp",
+    app_icons = ["//mediapipe/examples/ios/common:AppIcon"],
+    bundle_id = BUNDLE_ID_PREFIX + ".SelfieSegmentationGpu",
+    families = [
+        "iphone",
+        "ipad",
+    ],
+    infoplists = [
+        "//mediapipe/examples/ios/common:Info.plist",
+        "Info.plist",
+    ],
+    minimum_os_version = MIN_IOS_VERSION,
+    provisioning_profile = example_provisioning(),
+    deps = [
+        ":SelfieSegmentationGpuAppLibrary",
+        "@ios_opencv//:OpencvFramework",
+    ],
+)
+
+objc_library(
+    name = "SelfieSegmentationGpuAppLibrary",
+    data = [
+        "//mediapipe/graphs/selfie_segmentation:selfie_segmentation_gpu.binarypb",
+        "//mediapipe/modules/selfie_segmentation:selfie_segmentation.tflite",
+    ],
+    deps = [
+        "//mediapipe/examples/ios/common:CommonMediaPipeAppLibrary",
+    ] + select({
+        "//mediapipe:ios_i386": [],
+        "//mediapipe:ios_x86_64": [],
+        "//conditions:default": [
+            "//mediapipe/graphs/selfie_segmentation:selfie_segmentation_gpu_deps",
+        ],
+    }),
+)

mediapipe/examples/ios/selfiesegmentationgpu/Info.plist

@@ -0,0 +1,14 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+  <key>CameraPosition</key>
+  <string>front</string>
+  <key>GraphOutputStream</key>
+  <string>output_video</string>
+  <key>GraphInputStream</key>
+  <string>input_video</string>
+  <key>GraphName</key>
+  <string>selfie_segmentation_gpu</string>
+</dict>
+</plist>