78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
# Copyright 2023 The MediaPipe Authors. All Rights Reserved.
|
|
#
|
|
# 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(
|
|
"//mediapipe/framework/tool:ios.bzl",
|
|
"MPP_TASK_MINIMUM_OS_VERSION",
|
|
)
|
|
load(
|
|
"@org_tensorflow//tensorflow/lite:special_rules.bzl",
|
|
"tflite_ios_lab_runner",
|
|
)
|
|
load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test")
|
|
|
|
package(default_visibility = ["//visibility:public"])
|
|
|
|
licenses(["notice"])
|
|
|
|
# Default tags for filtering iOS targets. Targets are restricted to Apple platforms.
|
|
TFL_DEFAULT_TAGS = [
|
|
"apple",
|
|
]
|
|
|
|
# Following sanitizer tests are not supported by iOS test targets.
|
|
TFL_DISABLED_SANITIZER_TAGS = [
|
|
"noasan",
|
|
"nomsan",
|
|
"notsan",
|
|
]
|
|
|
|
objc_library(
|
|
name = "MPPImageSegmenterObjcTestLibrary",
|
|
testonly = 1,
|
|
srcs = ["MPPImageSegmenterTests.mm"],
|
|
copts = [
|
|
"-ObjC++",
|
|
"-std=c++17",
|
|
"-x objective-c++",
|
|
],
|
|
data = [
|
|
"//mediapipe/tasks/testdata/vision:test_images",
|
|
"//mediapipe/tasks/testdata/vision:test_models",
|
|
"//mediapipe/tasks/testdata/vision:test_protos",
|
|
],
|
|
deps = [
|
|
"//mediapipe/tasks/ios/common:MPPCommon",
|
|
"//mediapipe/tasks/ios/test/vision/utils:MPPImageTestUtils",
|
|
"//mediapipe/tasks/ios/test/vision/utils:MPPMaskTestUtils",
|
|
"//mediapipe/tasks/ios/vision/image_segmenter:MPPImageSegmenter",
|
|
"//mediapipe/tasks/ios/vision/image_segmenter:MPPImageSegmenterResult",
|
|
] + select({
|
|
"//third_party:opencv_ios_sim_arm64_source_build": ["@ios_opencv_source//:opencv_xcframework"],
|
|
"//third_party:opencv_ios_arm64_source_build": ["@ios_opencv_source//:opencv_xcframework"],
|
|
"//third_party:opencv_ios_x86_64_source_build": ["@ios_opencv_source//:opencv_xcframework"],
|
|
"//conditions:default": ["@ios_opencv//:OpencvFramework"],
|
|
}),
|
|
)
|
|
|
|
ios_unit_test(
|
|
name = "MPPImageSegmenterObjcTest",
|
|
minimum_os_version = MPP_TASK_MINIMUM_OS_VERSION,
|
|
runner = tflite_ios_lab_runner("IOS_LATEST"),
|
|
tags = TFL_DEFAULT_TAGS + TFL_DISABLED_SANITIZER_TAGS,
|
|
deps = [
|
|
":MPPImageSegmenterObjcTestLibrary",
|
|
],
|
|
)
|