experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add a test app for TFLite ObjC API.

Browse Source 
PiperOrigin-RevId: 249390128
This commit is contained in:
A. Unique TensorFlower 2019-05-21 22:57:55 -07:00 committed by TensorFlower Gardener
parent 731409f8d2
commit 2be59a5191
22 changed files with 1254 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
tensorflow/lite/experimental/objc/BUILD.apple
View File

@ -5,7 +5,7 @@ package(default_visibility = ["//visibility:private"])
licenses(["notice"]) # Apache 2.0
load("//tensorflow/lite/experimental/ios:ios.bzl", "TFL_DEFAULT_TAGS", "TFL_DISABLED_SANITIZER_TAGS", "TFL_MINIMUM_OS_VERSION")
load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test")
load("@build_bazel_rules_apple//apple:ios.bzl", "ios_application", "ios_unit_test")
SOURCES = glob([
"sources/*.h",
@ -58,7 +58,6 @@ objc_library(
srcs = SOURCES,
hdrs = API_HEADERS,
copts = RELEASE_COPTS,
module_map = "apis/module.modulemap",
tags = TFL_DEFAULT_TAGS,
deps = [
"//tensorflow/lite/experimental/c:c_api",
@ -67,17 +66,17 @@ objc_library(
)
ios_unit_test(
name = "TensorFlowLiteTests",
name = "Tests",
size = "medium",
minimum_os_version = TFL_MINIMUM_OS_VERSION,
tags = TFL_DEFAULT_TAGS + TFL_DISABLED_SANITIZER_TAGS,
deps = [
":TestsLib",
":TestsLibrary",
],
)
objc_library(
name = "TestsLib",
name = "TestsLibrary",
testonly = 1,
srcs = glob([
"tests/*.m",
@ -97,3 +96,41 @@ objc_library(
":TensorFlowLite",
],
)
ios_application(
name = "TestApp",
app_icons = glob(["apps/TestApp/TestApp/Assets.xcassets/AppIcon.appiconset/**"]),
bundle_id = "com.tensorflow.lite.objc.TestApp",
families = [
"ipad",
"iphone",
],
infoplists = ["apps/TestApp/TestApp/Info.plist"],
minimum_os_version = TFL_MINIMUM_OS_VERSION,
sdk_frameworks = [
"CoreGraphics",
],
tags = TFL_DEFAULT_TAGS,
deps = [
":TestAppLibrary",
],
)
objc_library(
name = "TestAppLibrary",
srcs = glob(["apps/TestApp/TestApp/*.m"]),
hdrs = glob(["apps/TestApp/TestApp/*.h"]),
data = glob(["apps/TestApp/TestApp/Base.lproj/*.storyboard"]) + [
"//tensorflow/lite:testdata/add.bin",
"//tensorflow/lite:testdata/add_quantized.bin",
"//tensorflow/lite:testdata/multi_add.bin",
],
includes = [
"apis",
],
module_name = "TestApp",
tags = TFL_DEFAULT_TAGS + ["manual"],
deps = [
":TensorFlowLite",
],
)

4
tensorflow/lite/experimental/objc/README.md
View File

@ -74,10 +74,10 @@ Build the `TensorFlowLite` Objective-C library target:
bazel build tensorflow/lite/experimental/objc:TensorFlowLite
```
Build the `TensorFlowLiteTests` target:
Build the `Tests` target:
```shell
bazel test tensorflow/lite/experimental/objc:TensorFlowLiteTests
bazel test tensorflow/lite/experimental/objc:Tests
```
#### Generate the Xcode project using Tulsi

5
tensorflow/lite/experimental/objc/TensorFlowLite.tulsiproj/Configs/TensorFlowLite.tulsigen
View File

@ -4,6 +4,8 @@
"tensorflow/lite/experimental/c",
"tensorflow/lite/experimental/objc",
"tensorflow/lite/experimental/objc/apis",
"tensorflow/lite/experimental/objc/apps/TestApp/TestApp",
"tensorflow/lite/experimental/objc/apps/TestApp/TestApp/Base.lproj",
"tensorflow/lite/experimental/objc/sources",
"tensorflow/lite/experimental/objc/tests",
"tensorflow/lite/kernels",
@ -13,7 +15,8 @@
],
"buildTargets" : [
"//tensorflow/lite/experimental/objc:TensorFlowLite",
"//tensorflow/lite/experimental/objc:TensorFlowLiteTests",
"//tensorflow/lite/experimental/objc:TestApp",
"//tensorflow/lite/experimental/objc:Tests",
],
"projectName" : "TensorFlowLite",
"optionSet" : {

5
tensorflow/lite/experimental/objc/apis/framework.modulemap
View File

@ -1,10 +1,5 @@
framework module TFLTensorFlowLite {
umbrella header "TFLTensorFlowLite.h"
header "TFLInterpreter.h"
header "TFLInterpreterOptions.h"
header "TFLQuantizationParameters.h"
header "TFLTensor.h"
export *
}

10
tensorflow/lite/experimental/objc/apis/module.modulemap
View File

@ -1,10 +0,0 @@
module TFLTensorFlowLite {
umbrella header "TFLTensorFlowLite.h"
header "TFLInterpreter.h"
header "TFLInterpreterOptions.h"
header "TFLQuantizationParameters.h"
header "TFLTensor.h"
export *
}

5
tensorflow/lite/experimental/objc/apps/TestApp/Podfile Normal file
View File

@ -0,0 +1,5 @@
platform :ios, '9.0'
target 'TestApp' do
pod 'TensorFlowLiteObjC'
end

352
tensorflow/lite/experimental/objc/apps/TestApp/TestApp.xcodeproj/project.pbxproj Normal file
View File

@ -0,0 +1,352 @@
// !$*UTF8*$!
{
archiveVersion = 1;
classes = {
};
objectVersion = 50;
objects = {
/* Begin PBXBuildFile section */
B20F5F3622937C9000A4FBD8 /* add_quantized.bin in Resources */ = {isa = PBXBuildFile; fileRef = B20F5F3322937C8D00A4FBD8 /* add_quantized.bin */; };
B20F5F3722937C9000A4FBD8 /* add.bin in Resources */ = {isa = PBXBuildFile; fileRef = B20F5F3422937C8F00A4FBD8 /* add.bin */; };
B20F5F3822937C9000A4FBD8 /* multi_add.bin in Resources */ = {isa = PBXBuildFile; fileRef = B20F5F3522937C8F00A4FBD8 /* multi_add.bin */; };
B210BD922291D78D00572163 /* AppDelegate.m in Sources */ = {isa = PBXBuildFile; fileRef = B210BD912291D78D00572163 /* AppDelegate.m */; };
B210BD952291D78D00572163 /* ViewController.m in Sources */ = {isa = PBXBuildFile; fileRef = B210BD942291D78D00572163 /* ViewController.m */; };
B210BD982291D78D00572163 /* Main.storyboard in Resources */ = {isa = PBXBuildFile; fileRef = B210BD962291D78D00572163 /* Main.storyboard */; };
B210BD9A2291D78E00572163 /* Assets.xcassets in Resources */ = {isa = PBXBuildFile; fileRef = B210BD992291D78E00572163 /* Assets.xcassets */; };
B210BD9D2291D78F00572163 /* LaunchScreen.storyboard in Resources */ = {isa = PBXBuildFile; fileRef = B210BD9B2291D78F00572163 /* LaunchScreen.storyboard */; };
B210BDA02291D78F00572163 /* main.m in Sources */ = {isa = PBXBuildFile; fileRef = B210BD9F2291D78F00572163 /* main.m */; };
/* End PBXBuildFile section */
/* Begin PBXFileReference section */
B20F5F3322937C8D00A4FBD8 /* add_quantized.bin */ = {isa = PBXFileReference; lastKnownFileType = archive.macbinary; name = add_quantized.bin; path = ../../../../testdata/add_quantized.bin; sourceTree = "<group>"; };
B20F5F3422937C8F00A4FBD8 /* add.bin */ = {isa = PBXFileReference; lastKnownFileType = archive.macbinary; name = add.bin; path = ../../../../testdata/add.bin; sourceTree = "<group>"; };
B20F5F3522937C8F00A4FBD8 /* multi_add.bin */ = {isa = PBXFileReference; lastKnownFileType = archive.macbinary; name = multi_add.bin; path = ../../../../testdata/multi_add.bin; sourceTree = "<group>"; };
B210BD8D2291D78D00572163 /* TestApp.app */ = {isa = PBXFileReference; explicitFileType = wrapper.application; includeInIndex = 0; path = TestApp.app; sourceTree = BUILT_PRODUCTS_DIR; };
B210BD902291D78D00572163 /* AppDelegate.h */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.c.h; path = AppDelegate.h; sourceTree = "<group>"; };
B210BD912291D78D00572163 /* AppDelegate.m */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.c.objc; path = AppDelegate.m; sourceTree = "<group>"; };
B210BD932291D78D00572163 /* ViewController.h */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.c.h; path = ViewController.h; sourceTree = "<group>"; };
B210BD942291D78D00572163 /* ViewController.m */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.c.objc; path = ViewController.m; sourceTree = "<group>"; };
B210BD972291D78D00572163 /* Base */ = {isa = PBXFileReference; lastKnownFileType = file.storyboard; name = Base; path = Base.lproj/Main.storyboard; sourceTree = "<group>"; };
B210BD992291D78E00572163 /* Assets.xcassets */ = {isa = PBXFileReference; lastKnownFileType = folder.assetcatalog; path = Assets.xcassets; sourceTree = "<group>"; };
B210BD9C2291D78F00572163 /* Base */ = {isa = PBXFileReference; lastKnownFileType = file.storyboard; name = Base; path = Base.lproj/LaunchScreen.storyboard; sourceTree = "<group>"; };
B210BD9E2291D78F00572163 /* Info.plist */ = {isa = PBXFileReference; lastKnownFileType = text.plist.xml; path = Info.plist; sourceTree = "<group>"; };
B210BD9F2291D78F00572163 /* main.m */ = {isa = PBXFileReference; lastKnownFileType = sourcecode.c.objc; path = main.m; sourceTree = "<group>"; };
/* End PBXFileReference section */
/* Begin PBXFrameworksBuildPhase section */
B210BD8A2291D78D00572163 /* Frameworks */ = {
isa = PBXFrameworksBuildPhase;
buildActionMask = 2147483647;
files = (
);
runOnlyForDeploymentPostprocessing = 0;
};
/* End PBXFrameworksBuildPhase section */
/* Begin PBXGroup section */
B210BD842291D78D00572163 = {
isa = PBXGroup;
children = (
B20F5F3322937C8D00A4FBD8 /* add_quantized.bin */,
B20F5F3422937C8F00A4FBD8 /* add.bin */,
B20F5F3522937C8F00A4FBD8 /* multi_add.bin */,
B210BD8F2291D78D00572163 /* TestApp */,
B210BD8E2291D78D00572163 /* Products */,
);
sourceTree = "<group>";
};
B210BD8E2291D78D00572163 /* Products */ = {
isa = PBXGroup;
children = (
B210BD8D2291D78D00572163 /* TestApp.app */,
);
name = Products;
sourceTree = "<group>";
};
B210BD8F2291D78D00572163 /* TestApp */ = {
isa = PBXGroup;
children = (
B210BD902291D78D00572163 /* AppDelegate.h */,
B210BD912291D78D00572163 /* AppDelegate.m */,
B210BD932291D78D00572163 /* ViewController.h */,
B210BD942291D78D00572163 /* ViewController.m */,
B210BD962291D78D00572163 /* Main.storyboard */,
B210BD992291D78E00572163 /* Assets.xcassets */,
B210BD9B2291D78F00572163 /* LaunchScreen.storyboard */,
B210BD9E2291D78F00572163 /* Info.plist */,
B210BD9F2291D78F00572163 /* main.m */,
);
path = TestApp;
sourceTree = "<group>";
};
/* End PBXGroup section */
/* Begin PBXNativeTarget section */
B210BD8C2291D78D00572163 /* TestApp */ = {
isa = PBXNativeTarget;
buildConfigurationList = B210BDA32291D78F00572163 /* Build configuration list for PBXNativeTarget "TestApp" */;
buildPhases = (
B210BD892291D78D00572163 /* Sources */,
B210BD8A2291D78D00572163 /* Frameworks */,
B210BD8B2291D78D00572163 /* Resources */,
);
buildRules = (
);
dependencies = (
);
name = TestApp;
productName = TestApp;
productReference = B210BD8D2291D78D00572163 /* TestApp.app */;
productType = "com.apple.product-type.application";
};
/* End PBXNativeTarget section */
/* Begin PBXProject section */
B210BD852291D78D00572163 /* Project object */ = {
isa = PBXProject;
attributes = {
LastUpgradeCheck = 1010;
ORGANIZATIONNAME = "Google Inc";
TargetAttributes = {
B210BD8C2291D78D00572163 = {
CreatedOnToolsVersion = 10.1;
};
};
};
buildConfigurationList = B210BD882291D78D00572163 /* Build configuration list for PBXProject "TestApp" */;
compatibilityVersion = "Xcode 9.3";
developmentRegion = en;
hasScannedForEncodings = 0;
knownRegions = (
en,
Base,
);
mainGroup = B210BD842291D78D00572163;
productRefGroup = B210BD8E2291D78D00572163 /* Products */;
projectDirPath = "";
projectRoot = "";
targets = (
B210BD8C2291D78D00572163 /* TestApp */,
);
};
/* End PBXProject section */
/* Begin PBXResourcesBuildPhase section */
B210BD8B2291D78D00572163 /* Resources */ = {
isa = PBXResourcesBuildPhase;
buildActionMask = 2147483647;
files = (
B210BD9D2291D78F00572163 /* LaunchScreen.storyboard in Resources */,
B20F5F3622937C9000A4FBD8 /* add_quantized.bin in Resources */,
B20F5F3722937C9000A4FBD8 /* add.bin in Resources */,
B210BD9A2291D78E00572163 /* Assets.xcassets in Resources */,
B210BD982291D78D00572163 /* Main.storyboard in Resources */,
B20F5F3822937C9000A4FBD8 /* multi_add.bin in Resources */,
);
runOnlyForDeploymentPostprocessing = 0;
};
/* End PBXResourcesBuildPhase section */
/* Begin PBXSourcesBuildPhase section */
B210BD892291D78D00572163 /* Sources */ = {
isa = PBXSourcesBuildPhase;
buildActionMask = 2147483647;
files = (
B210BD952291D78D00572163 /* ViewController.m in Sources */,
B210BDA02291D78F00572163 /* main.m in Sources */,
B210BD922291D78D00572163 /* AppDelegate.m in Sources */,
);
runOnlyForDeploymentPostprocessing = 0;
};
/* End PBXSourcesBuildPhase section */
/* Begin PBXVariantGroup section */
B210BD962291D78D00572163 /* Main.storyboard */ = {
isa = PBXVariantGroup;
children = (
B210BD972291D78D00572163 /* Base */,
);
name = Main.storyboard;
sourceTree = "<group>";
};
B210BD9B2291D78F00572163 /* LaunchScreen.storyboard */ = {
isa = PBXVariantGroup;
children = (
B210BD9C2291D78F00572163 /* Base */,
);
name = LaunchScreen.storyboard;
sourceTree = "<group>";
};
/* End PBXVariantGroup section */
/* Begin XCBuildConfiguration section */
B210BDA12291D78F00572163 /* Debug */ = {
isa = XCBuildConfiguration;
buildSettings = {
ALWAYS_SEARCH_USER_PATHS = NO;
CLANG_ANALYZER_NONNULL = YES;
CLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;
CLANG_CXX_LANGUAGE_STANDARD = "gnu++14";
CLANG_CXX_LIBRARY = "libc++";
CLANG_ENABLE_MODULES = YES;
CLANG_ENABLE_OBJC_ARC = YES;
CLANG_ENABLE_OBJC_WEAK = YES;
CLANG_WARN_BLOCK_CAPTURE_AUTORELEASING = YES;
CLANG_WARN_BOOL_CONVERSION = YES;
CLANG_WARN_COMMA = YES;
CLANG_WARN_CONSTANT_CONVERSION = YES;
CLANG_WARN_DEPRECATED_OBJC_IMPLEMENTATIONS = YES;
CLANG_WARN_DIRECT_OBJC_ISA_USAGE = YES_ERROR;
CLANG_WARN_DOCUMENTATION_COMMENTS = YES;
CLANG_WARN_EMPTY_BODY = YES;
CLANG_WARN_ENUM_CONVERSION = YES;
CLANG_WARN_INFINITE_RECURSION = YES;
CLANG_WARN_INT_CONVERSION = YES;
CLANG_WARN_NON_LITERAL_NULL_CONVERSION = YES;
CLANG_WARN_OBJC_IMPLICIT_RETAIN_SELF = YES;
CLANG_WARN_OBJC_LITERAL_CONVERSION = YES;
CLANG_WARN_OBJC_ROOT_CLASS = YES_ERROR;
CLANG_WARN_RANGE_LOOP_ANALYSIS = YES;
CLANG_WARN_STRICT_PROTOTYPES = YES;
CLANG_WARN_SUSPICIOUS_MOVE = YES;
CLANG_WARN_UNGUARDED_AVAILABILITY = YES_AGGRESSIVE;
CLANG_WARN_UNREACHABLE_CODE = YES;
CLANG_WARN__DUPLICATE_METHOD_MATCH = YES;
CODE_SIGN_IDENTITY = "iPhone Developer";
COPY_PHASE_STRIP = NO;
DEBUG_INFORMATION_FORMAT = dwarf;
ENABLE_STRICT_OBJC_MSGSEND = YES;
ENABLE_TESTABILITY = YES;
GCC_C_LANGUAGE_STANDARD = gnu11;
GCC_DYNAMIC_NO_PIC = NO;
GCC_NO_COMMON_BLOCKS = YES;
GCC_OPTIMIZATION_LEVEL = 0;
GCC_PREPROCESSOR_DEFINITIONS = (
"DEBUG=1",
"$(inherited)",
);
GCC_WARN_64_TO_32_BIT_CONVERSION = YES;
GCC_WARN_ABOUT_RETURN_TYPE = YES_ERROR;
GCC_WARN_UNDECLARED_SELECTOR = YES;
GCC_WARN_UNINITIALIZED_AUTOS = YES_AGGRESSIVE;
GCC_WARN_UNUSED_FUNCTION = YES;
GCC_WARN_UNUSED_VARIABLE = YES;
IPHONEOS_DEPLOYMENT_TARGET = 12.1;
MTL_ENABLE_DEBUG_INFO = INCLUDE_SOURCE;
MTL_FAST_MATH = YES;
ONLY_ACTIVE_ARCH = YES;
SDKROOT = iphoneos;
};
name = Debug;
};
B210BDA22291D78F00572163 /* Release */ = {
isa = XCBuildConfiguration;
buildSettings = {
ALWAYS_SEARCH_USER_PATHS = NO;
CLANG_ANALYZER_NONNULL = YES;
CLANG_ANALYZER_NUMBER_OBJECT_CONVERSION = YES_AGGRESSIVE;
CLANG_CXX_LANGUAGE_STANDARD = "gnu++14";
CLANG_CXX_LIBRARY = "libc++";
CLANG_ENABLE_MODULES = YES;
CLANG_ENABLE_OBJC_ARC = YES;
CLANG_ENABLE_OBJC_WEAK = YES;
CLANG_WARN_BLOCK_CAPTURE_AUTORELEASING = YES;
CLANG_WARN_BOOL_CONVERSION = YES;
CLANG_WARN_COMMA = YES;
CLANG_WARN_CONSTANT_CONVERSION = YES;
CLANG_WARN_DEPRECATED_OBJC_IMPLEMENTATIONS = YES;
CLANG_WARN_DIRECT_OBJC_ISA_USAGE = YES_ERROR;
CLANG_WARN_DOCUMENTATION_COMMENTS = YES;
CLANG_WARN_EMPTY_BODY = YES;
CLANG_WARN_ENUM_CONVERSION = YES;
CLANG_WARN_INFINITE_RECURSION = YES;
CLANG_WARN_INT_CONVERSION = YES;
CLANG_WARN_NON_LITERAL_NULL_CONVERSION = YES;
CLANG_WARN_OBJC_IMPLICIT_RETAIN_SELF = YES;
CLANG_WARN_OBJC_LITERAL_CONVERSION = YES;
CLANG_WARN_OBJC_ROOT_CLASS = YES_ERROR;
CLANG_WARN_RANGE_LOOP_ANALYSIS = YES;
CLANG_WARN_STRICT_PROTOTYPES = YES;
CLANG_WARN_SUSPICIOUS_MOVE = YES;
CLANG_WARN_UNGUARDED_AVAILABILITY = YES_AGGRESSIVE;
CLANG_WARN_UNREACHABLE_CODE = YES;
CLANG_WARN__DUPLICATE_METHOD_MATCH = YES;
CODE_SIGN_IDENTITY = "iPhone Developer";
COPY_PHASE_STRIP = NO;
DEBUG_INFORMATION_FORMAT = "dwarf-with-dsym";
ENABLE_NS_ASSERTIONS = NO;
ENABLE_STRICT_OBJC_MSGSEND = YES;
GCC_C_LANGUAGE_STANDARD = gnu11;
GCC_NO_COMMON_BLOCKS = YES;
GCC_WARN_64_TO_32_BIT_CONVERSION = YES;
GCC_WARN_ABOUT_RETURN_TYPE = YES_ERROR;
GCC_WARN_UNDECLARED_SELECTOR = YES;
GCC_WARN_UNINITIALIZED_AUTOS = YES_AGGRESSIVE;
GCC_WARN_UNUSED_FUNCTION = YES;
GCC_WARN_UNUSED_VARIABLE = YES;
IPHONEOS_DEPLOYMENT_TARGET = 12.1;
MTL_ENABLE_DEBUG_INFO = NO;
MTL_FAST_MATH = YES;
SDKROOT = iphoneos;
VALIDATE_PRODUCT = YES;
};
name = Release;
};
B210BDA42291D78F00572163 /* Debug */ = {
isa = XCBuildConfiguration;
buildSettings = {
ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
CODE_SIGN_STYLE = Automatic;
INFOPLIST_FILE = TestApp/Info.plist;
LD_RUNPATH_SEARCH_PATHS = (
"$(inherited)",
"@executable_path/Frameworks",
);
PRODUCT_BUNDLE_IDENTIFIER = com.tensorflow.lite.objc.TestApp;
PRODUCT_NAME = "$(TARGET_NAME)";
TARGETED_DEVICE_FAMILY = "1,2";
};
name = Debug;
};
B210BDA52291D78F00572163 /* Release */ = {
isa = XCBuildConfiguration;
buildSettings = {
ASSETCATALOG_COMPILER_APPICON_NAME = AppIcon;
CODE_SIGN_STYLE = Automatic;
INFOPLIST_FILE = TestApp/Info.plist;
LD_RUNPATH_SEARCH_PATHS = (
"$(inherited)",
"@executable_path/Frameworks",
);
PRODUCT_BUNDLE_IDENTIFIER = com.tensorflow.lite.objc.TestApp;
PRODUCT_NAME = "$(TARGET_NAME)";
TARGETED_DEVICE_FAMILY = "1,2";
};
name = Release;
};
/* End XCBuildConfiguration section */
/* Begin XCConfigurationList section */
B210BD882291D78D00572163 /* Build configuration list for PBXProject "TestApp" */ = {
isa = XCConfigurationList;
buildConfigurations = (
B210BDA12291D78F00572163 /* Debug */,
B210BDA22291D78F00572163 /* Release */,
);
defaultConfigurationIsVisible = 0;
defaultConfigurationName = Release;
};
B210BDA32291D78F00572163 /* Build configuration list for PBXNativeTarget "TestApp" */ = {
isa = XCConfigurationList;
buildConfigurations = (
B210BDA42291D78F00572163 /* Debug */,
B210BDA52291D78F00572163 /* Release */,
);
defaultConfigurationIsVisible = 0;
defaultConfigurationName = Release;
};
/* End XCConfigurationList section */
};
rootObject = B210BD852291D78D00572163 /* Project object */;
}

25
tensorflow/lite/experimental/objc/apps/TestApp/TestApp/AppDelegate.h Normal file
View File

@ -0,0 +1,25 @@
// Copyright 2019 Google Inc. 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.
#import <UIKit/UIKit.h>
NS_ASSUME_NONNULL_BEGIN
@interface AppDelegate : UIResponder <UIApplicationDelegate>
@property(nonatomic) UIWindow *window;
@end
NS_ASSUME_NONNULL_END

22
tensorflow/lite/experimental/objc/apps/TestApp/TestApp/AppDelegate.m Normal file
View File

@ -0,0 +1,22 @@
// Copyright 2019 Google Inc. 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.
#import "AppDelegate.h"
NS_ASSUME_NONNULL_BEGIN
@implementation AppDelegate
@end
NS_ASSUME_NONNULL_END

98
tensorflow/lite/experimental/objc/apps/TestApp/TestApp/Assets.xcassets/AppIcon.appiconset/Contents.json Normal file
View File

@ -0,0 +1,98 @@
{
"images" : [
{
"idiom" : "iphone",
"size" : "20x20",
"scale" : "2x"
},
{
"idiom" : "iphone",
"size" : "20x20",
"scale" : "3x"
},
{
"idiom" : "iphone",
"size" : "29x29",
"scale" : "2x"
},
{
"idiom" : "iphone",
"size" : "29x29",
"scale" : "3x"
},
{
"idiom" : "iphone",
"size" : "40x40",
"scale" : "2x"
},
{
"idiom" : "iphone",
"size" : "40x40",
"scale" : "3x"
},
{
"idiom" : "iphone",
"size" : "60x60",
"scale" : "2x"
},
{
"idiom" : "iphone",
"size" : "60x60",
"scale" : "3x"
},
{
"idiom" : "ipad",
"size" : "20x20",
"scale" : "1x"
},
{
"idiom" : "ipad",
"size" : "20x20",
"scale" : "2x"
},
{
"idiom" : "ipad",
"size" : "29x29",
"scale" : "1x"
},
{
"idiom" : "ipad",
"size" : "29x29",
"scale" : "2x"
},
{
"idiom" : "ipad",
"size" : "40x40",
"scale" : "1x"
},
{
"idiom" : "ipad",
"size" : "40x40",
"scale" : "2x"
},
{
"idiom" : "ipad",
"size" : "76x76",
"scale" : "1x"
},
{
"idiom" : "ipad",
"size" : "76x76",
"scale" : "2x"
},
{
"idiom" : "ipad",
"size" : "83.5x83.5",
"scale" : "2x"
},
{
"idiom" : "ios-marketing",
"size" : "1024x1024",
"scale" : "1x"
}
],
"info" : {
"version" : 1,
"author" : "xcode"
}
}

6
tensorflow/lite/experimental/objc/apps/TestApp/TestApp/Assets.xcassets/Contents.json Normal file
View File

@ -0,0 +1,6 @@
{
"info" : {
"version" : 1,
"author" : "xcode"
}
}

44
tensorflow/lite/experimental/objc/apps/TestApp/TestApp/Base.lproj/LaunchScreen.storyboard Normal file
View File

@ -0,0 +1,44 @@
<?xml version="1.0" encoding="UTF-8"?>
<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="14460.31" targetRuntime="iOS.CocoaTouch" propertyAccessControl="none" useAutolayout="YES" launchScreen="YES" useTraitCollections="YES" colorMatched="YES" initialViewController="01J-lp-oVM">
<device id="retina4_7" orientation="portrait">
<adaptation id="fullscreen"/>
</device>
<dependencies>
<deployment identifier="iOS"/>
<plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="14460.20"/>
<capability name="documents saved in the Xcode 8 format" minToolsVersion="8.0"/>
</dependencies>
<scenes>
<!--View Controller-->
<scene sceneID="EHf-IW-A2E">
<objects>
<viewController id="01J-lp-oVM" sceneMemberID="viewController">
<layoutGuides>
<viewControllerLayoutGuide type="top" id="Y0Z-8F-bB8"/>
<viewControllerLayoutGuide type="bottom" id="TqJ-Hq-gHs"/>
</layoutGuides>
<view key="view" contentMode="scaleToFill" id="Ze5-6b-2t3">
<rect key="frame" x="0.0" y="0.0" width="375" height="667"/>
<autoresizingMask key="autoresizingMask" widthSizable="YES" heightSizable="YES"/>
<subviews>
<label opaque="NO" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="TensorFlow Lite Test" textAlignment="center" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" minimumScaleFactor="0.25" translatesAutoresizingMaskIntoConstraints="NO" id="zIC-MS-HeK">
<rect key="frame" x="16" y="314" width="343" height="39"/>
<fontDescription key="fontDescription" type="boldSystem" pointSize="32"/>
<color key="textColor" red="1" green="0.50329624702372611" blue="0.013296667412401542" alpha="0.84705882352941175" colorSpace="custom" customColorSpace="displayP3"/>
<nil key="highlightedColor"/>
</label>
</subviews>
<color key="backgroundColor" red="1" green="1" blue="1" alpha="1" colorSpace="custom" customColorSpace="sRGB"/>
<constraints>
<constraint firstAttribute="trailing" secondItem="zIC-MS-HeK" secondAttribute="trailing" constant="16" id="1Wp-jb-ol6"/>
<constraint firstItem="zIC-MS-HeK" firstAttribute="centerY" secondItem="Ze5-6b-2t3" secondAttribute="centerY" id="R2T-Hp-TBa"/>
<constraint firstItem="zIC-MS-HeK" firstAttribute="leading" secondItem="Ze5-6b-2t3" secondAttribute="leading" constant="16" id="alE-O6-WL6"/>
</constraints>
</view>
</viewController>
<placeholder placeholderIdentifier="IBFirstResponder" id="iYj-Kq-Ea1" userLabel="First Responder" sceneMemberID="firstResponder"/>
</objects>
<point key="canvasLocation" x="52" y="374.66266866566718"/>
</scene>
</scenes>
</document>

89
tensorflow/lite/experimental/objc/apps/TestApp/TestApp/Base.lproj/Main.storyboard Normal file
View File

@ -0,0 +1,89 @@
<?xml version="1.0" encoding="UTF-8"?>
<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="14460.31" targetRuntime="iOS.CocoaTouch" propertyAccessControl="none" useAutolayout="YES" useTraitCollections="YES" colorMatched="YES" initialViewController="BYZ-38-t0r">
<device id="retina4_7" orientation="portrait">
<adaptation id="fullscreen"/>
</device>
<dependencies>
<deployment version="2304" identifier="iOS"/>
<plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="14460.20"/>
<capability name="documents saved in the Xcode 8 format" minToolsVersion="8.0"/>
</dependencies>
<scenes>
<!--View Controller-->
<scene sceneID="tne-QT-ifu">
<objects>
<viewController id="BYZ-38-t0r" customClass="ViewController" sceneMemberID="viewController">
<layoutGuides>
<viewControllerLayoutGuide type="top" id="5Qd-iD-SiH"/>
<viewControllerLayoutGuide type="bottom" id="zTS-de-uge"/>
</layoutGuides>
<view key="view" contentMode="scaleToFill" id="8bC-Xf-vdC">
<rect key="frame" x="0.0" y="0.0" width="375" height="667"/>
<autoresizingMask key="autoresizingMask" widthSizable="YES" heightSizable="YES"/>
<subviews>
<toolbar opaque="NO" clearsContextBeforeDrawing="NO" contentMode="scaleToFill" translatesAutoresizingMaskIntoConstraints="NO" id="ahi-i4-2FP" userLabel="Top Model Toolbar">
<rect key="frame" x="0.0" y="34" width="375" height="44"/>
<constraints>
<constraint firstAttribute="height" constant="44" id="nwy-nk-0wZ"/>
</constraints>
<items>
<barButtonItem style="plain" id="Ywd-KS-s96">
<segmentedControl key="customView" opaque="NO" contentMode="scaleToFill" contentHorizontalAlignment="left" contentVerticalAlignment="top" segmentControlStyle="bar" selectedSegmentIndex="0" id="8kc-88-CHj" userLabel="Model Control">
<rect key="frame" x="16" y="7.5" width="343" height="29"/>
<autoresizingMask key="autoresizingMask" flexibleMaxX="YES" flexibleMaxY="YES"/>
<segments>
<segment title="Add"/>
<segment title="AddQuantized"/>
<segment title="MultiAdd"/>
</segments>
<connections>
<action selector="modelChanged:" destination="BYZ-38-t0r" eventType="valueChanged" id="z13-8K-EwC"/>
</connections>
</segmentedControl>
</barButtonItem>
</items>
</toolbar>
<toolbar opaque="NO" clearsContextBeforeDrawing="NO" contentMode="scaleToFill" translatesAutoresizingMaskIntoConstraints="NO" id="UWb-3E-O5r" userLabel="Bottom Invoke Toolbar">
<rect key="frame" x="0.0" y="140" width="375" height="44"/>
<items>
<barButtonItem title="Invoke Interpreter" width="374" id="He4-7G-biW">
<connections>
<action selector="invokeInterpreter:" destination="BYZ-38-t0r" id="Ycs-E9-Vul"/>
</connections>
</barButtonItem>
</items>
</toolbar>
<textView clipsSubviews="YES" multipleTouchEnabled="YES" contentMode="scaleToFill" misplaced="YES" editable="NO" adjustsFontForContentSizeCategory="YES" selectable="NO" translatesAutoresizingMaskIntoConstraints="NO" id="7Ws-3t-76I">
<rect key="frame" x="0.0" y="194" width="375" height="488"/>
<color key="backgroundColor" red="0.12820077356385221" green="0.40366933178860925" blue="0.96080166101455688" alpha="1" colorSpace="custom" customColorSpace="displayP3"/>
<color key="textColor" cocoaTouchSystemColor="tableCellGroupedBackgroundColor"/>
<fontDescription key="fontDescription" type="system" pointSize="14"/>
<textInputTraits key="textInputTraits" autocapitalizationType="sentences"/>
</textView>
</subviews>
<color key="backgroundColor" red="1" green="1" blue="1" alpha="1" colorSpace="custom" customColorSpace="sRGB"/>
<constraints>
<constraint firstItem="ahi-i4-2FP" firstAttribute="top" secondItem="5Qd-iD-SiH" secondAttribute="bottom" constant="14" id="0V2-16-9cM"/>
<constraint firstAttribute="trailing" secondItem="ahi-i4-2FP" secondAttribute="trailing" id="1D2-FC-OQ0"/>
<constraint firstItem="UWb-3E-O5r" firstAttribute="leading" secondItem="8bC-Xf-vdC" secondAttribute="leading" id="1To-8n-Knb"/>
<constraint firstItem="7Ws-3t-76I" firstAttribute="leading" secondItem="8bC-Xf-vdC" secondAttribute="leading" id="3Et-px-WCV"/>
<constraint firstItem="zTS-de-uge" firstAttribute="top" secondItem="7Ws-3t-76I" secondAttribute="bottom" id="Lkb-XF-ldX"/>
<constraint firstItem="7Ws-3t-76I" firstAttribute="top" secondItem="UWb-3E-O5r" secondAttribute="bottom" id="bXr-pF-Ld2"/>
<constraint firstItem="ahi-i4-2FP" firstAttribute="leading" secondItem="8bC-Xf-vdC" secondAttribute="leading" id="c98-HO-3m5"/>
<constraint firstAttribute="trailing" secondItem="UWb-3E-O5r" secondAttribute="trailing" id="oJz-hf-dJa"/>
<constraint firstAttribute="trailing" secondItem="7Ws-3t-76I" secondAttribute="trailing" id="oyy-C7-mUJ"/>
<constraint firstItem="UWb-3E-O5r" firstAttribute="top" secondItem="ahi-i4-2FP" secondAttribute="bottom" constant="62" id="uFg-MF-aJz"/>
</constraints>
</view>
<connections>
<outlet property="invokeButton" destination="He4-7G-biW" id="kpj-CS-Fss"/>
<outlet property="modelControl" destination="8kc-88-CHj" id="GTB-WG-ozW"/>
<outlet property="resultsTextView" destination="7Ws-3t-76I" id="fnd-i5-Pdh"/>
</connections>
</viewController>
<placeholder placeholderIdentifier="IBFirstResponder" id="dkx-z0-nzr" sceneMemberID="firstResponder"/>
</objects>
<point key="canvasLocation" x="136.80000000000001" y="132.68365817091455"/>
</scene>
</scenes>
</document>

42
tensorflow/lite/experimental/objc/apps/TestApp/TestApp/Info.plist Normal file
View File

@ -0,0 +1,42 @@
<?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>CFBundleDevelopmentRegion</key>
<string>en</string>
<key>CFBundleExecutable</key>
<string>$(EXECUTABLE_NAME)</string>
<key>CFBundleIdentifier</key>
<string>$(PRODUCT_BUNDLE_IDENTIFIER)</string>
<key>CFBundleInfoDictionaryVersion</key>
<string>6.0</string>
<key>CFBundleName</key>
<string>$(PRODUCT_NAME)</string>
<key>CFBundlePackageType</key>
<string>APPL</string>
<key>CFBundleShortVersionString</key>
<string>1.0</string>
<key>CFBundleVersion</key>
<string>0.0.1</string>
<key>LSRequiresIPhoneOS</key>
<true/>
<key>UILaunchStoryboardName</key>
<string>LaunchScreen</string>
<key>UIMainStoryboardFile</key>
<string>Main</string>
<key>UIRequiredDeviceCapabilities</key>
<array>
<string>armv7</string>
</array>
<key>UISupportedInterfaceOrientations</key>
<array>
<string>UIInterfaceOrientationPortrait</string>
<string>UIInterfaceOrientationPortraitUpsideDown</string>
</array>
<key>UISupportedInterfaceOrientations~ipad</key>
<array>
<string>UIInterfaceOrientationPortrait</string>
<string>UIInterfaceOrientationPortraitUpsideDown</string>
</array>
</dict>
</plist>

22
tensorflow/lite/experimental/objc/apps/TestApp/TestApp/ViewController.h Normal file
View File

@ -0,0 +1,22 @@
// Copyright 2019 Google Inc. 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.
#import <UIKit/UIKit.h>
NS_ASSUME_NONNULL_BEGIN
@interface ViewController : UIViewController
@end
NS_ASSUME_NONNULL_END

419
tensorflow/lite/experimental/objc/apps/TestApp/TestApp/ViewController.m Normal file
View File

@ -0,0 +1,419 @@
// Copyright 2019 Google Inc. 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.
#import "ViewController.h"
#import "TFLTensorFlowLite.h"
NS_ASSUME_NONNULL_BEGIN
/**
* Safely dispatches the given `block` on the main thread. If already on the main thread, the given
* block is executed immediately; otherwise, dispatches the block asynchronously on the main thread.
*
* @param block The block to dispatch on the main thread.
*/
void TLTSafeDispatchOnMain(dispatch_block_t block) {
if (block == nil) return;
if (NSThread.isMainThread) {
block();
} else {
dispatch_async(dispatch_get_main_queue(), block);
}
}
/**
* Name of a float model that performs two add operations on one input tensor and returns the result
* in one output tensor.
*/
static NSString *const kModelNameAdd = @"add";
/**
* Name of a quantized model that performs two add operations on one input tensor and returns the
* result in one output tensor.
*/
static NSString *const kModelNameAddQuantized = @"add_quantized";
/**
* Name of a float model that performs three add operations on four input tensors and returns the
* results in 2 output tensors.
*/
static NSString *const kModelNameMultiAdd = @"multi_add";
/** Model resource type. */
static NSString *const kModelType = @"bin";
/** The label for the serial queue for synchronizing interpreter calls. */
static const char *kInterpreterSerialQueueLabel = "com.tensorflow.lite.objc.testapp.interpreter";
static NSString *const kNilInterpreterError =
@"Failed to invoke the interpreter because the interpreter was nil.";
static NSString *const kInvokeInterpreterError = @"Failed to invoke interpreter due to error: %@.";
/** Model paths. */
static NSArray *gModelPaths;
@interface ViewController ()
/** Serial queue for synchronizing interpreter calls. */
@property(nonatomic) dispatch_queue_t interpreterSerialQueue;
/** TensorFlow Lite interpreter for the currently selected model. */
@property(nonatomic) TFLInterpreter *interpreter;
@property(weak, nonatomic) IBOutlet UISegmentedControl *modelControl;
@property(weak, nonatomic) IBOutlet UIBarButtonItem *invokeButton;
@property(weak, nonatomic) IBOutlet UITextView *resultsTextView;
@end
@implementation ViewController
#pragma mark - NSObject
+ (void)initialize {
if (self == [ViewController self]) {
gModelPaths = @[
[NSBundle.mainBundle pathForResource:kModelNameAdd ofType:kModelType],
[NSBundle.mainBundle pathForResource:kModelNameAddQuantized ofType:kModelType],
[NSBundle.mainBundle pathForResource:kModelNameMultiAdd ofType:kModelType],
];
}
}
#pragma mark - UIViewController
- (void)viewDidLoad {
[super viewDidLoad];
self.interpreterSerialQueue =
dispatch_queue_create(kInterpreterSerialQueueLabel, DISPATCH_QUEUE_SERIAL);
self.invokeButton.enabled = NO;
[self updateResultsText:[NSString stringWithFormat:@"Using TensorFlow Lite runtime version %@.",
TFLVersion]];
[self loadModel];
}
#pragma mark - IBActions
- (IBAction)modelChanged:(id)sender {
self.invokeButton.enabled = NO;
NSString *results = [NSString
stringWithFormat:@"Switched to the %@ model.",
[self.modelControl
titleForSegmentAtIndex:self.modelControl.selectedSegmentIndex]];
[self updateResultsText:results];
[self loadModel];
}
- (IBAction)invokeInterpreter:(id)sender {
switch (self.modelControl.selectedSegmentIndex) {
case 0:
[self invokeAdd];
break;
case 1:
[self invokeAddQuantized];
break;
case 2:
[self invokeMultiAdd];
}
}
#pragma mark - Private
/** Path of the currently selected model. */
- (nullable NSString *)currentModelPath {
return self.modelControl.selectedSegmentIndex == UISegmentedControlNoSegment
? nil
: gModelPaths[self.modelControl.selectedSegmentIndex];
}
- (void)loadModel {
NSString *modelPath = [self currentModelPath];
if (modelPath.length == 0) {
[self updateResultsText:@"No model is selected."];
return;
}
__weak typeof(self) weakSelf = self;
dispatch_async(self.interpreterSerialQueue, ^{
TFLInterpreterOptions *options = [[TFLInterpreterOptions alloc] init];
options.numberOfThreads = 2;
NSError *error;
weakSelf.interpreter = [[TFLInterpreter alloc] initWithModelPath:modelPath
options:options
error:&error];
if (weakSelf.interpreter == nil || error != nil) {
NSString *results =
[NSString stringWithFormat:@"Failed to create the interpreter due to error:%@",
error.localizedDescription];
[weakSelf updateResultsText:results];
} else {
TLTSafeDispatchOnMain(^{
weakSelf.invokeButton.enabled = YES;
});
}
});
}
- (void)invokeAdd {
__weak typeof(self) weakSelf = self;
dispatch_async(self.interpreterSerialQueue, ^{
if (weakSelf.interpreter == nil) {
[weakSelf updateResultsText:kNilInterpreterError];
return;
}
NSArray<NSNumber *> *shape = @[@2];
NSError *error;
if (![weakSelf.interpreter resizeInputTensorAtIndex:0 toShape:shape error:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
if (![weakSelf.interpreter allocateTensorsWithError:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
TFLTensor *inputTensor = [weakSelf.interpreter inputTensorAtIndex:0 error:&error];
if (inputTensor == nil || error != nil) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
NSMutableData *inputData = [NSMutableData dataWithCapacity:0];
float one = 1.f;
float three = 3.f;
[inputData appendBytes:&one length:sizeof(float)];
[inputData appendBytes:&three length:sizeof(float)];
if (![inputTensor copyData:inputData error:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
if (![weakSelf.interpreter invokeWithError:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
TFLTensor *outputTensor = [weakSelf.interpreter outputTensorAtIndex:0 error:&error];
if (outputTensor == nil || error != nil) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
NSData *outputData = [outputTensor dataWithError:&error];
if (outputData == nil || error != nil) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
float output[2U];
[outputData getBytes:output length:(sizeof(float) * 2U)];
[weakSelf
updateResultsText:[NSString stringWithFormat:@"Performing 2 add operations:\n\nInput = "
@"[%.1f, %.1f]\n\nOutput = [%.1f, %.1f]",
one, three, output[0], output[1]]];
});
}
- (void)invokeAddQuantized {
__weak typeof(self) weakSelf = self;
dispatch_async(self.interpreterSerialQueue, ^{
if (weakSelf.interpreter == nil) {
[weakSelf updateResultsText:kNilInterpreterError];
return;
}
NSArray<NSNumber *> *shape = @[@2];
NSError *error;
if (![weakSelf.interpreter resizeInputTensorAtIndex:0 toShape:shape error:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
if (![weakSelf.interpreter allocateTensorsWithError:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
TFLTensor *inputTensor = [weakSelf.interpreter inputTensorAtIndex:0 error:&error];
if (inputTensor == nil || error != nil) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
NSMutableData *inputData = [NSMutableData dataWithCapacity:0];
uint8_t one = 1U;
uint8_t three = 3U;
[inputData appendBytes:&one length:sizeof(uint8_t)];
[inputData appendBytes:&three length:sizeof(uint8_t)];
if (![inputTensor copyData:inputData error:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
if (![weakSelf.interpreter invokeWithError:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
TFLTensor *outputTensor = [weakSelf.interpreter outputTensorAtIndex:0 error:&error];
if (outputTensor == nil || error != nil) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
TFLQuantizationParameters *params = outputTensor.quantizationParameters;
if (params == nil) {
[weakSelf updateResultsText:
[NSString stringWithFormat:kInvokeInterpreterError,
@"Missing qualitization parameters in the output"]];
return;
}
NSData *outputData = [outputTensor dataWithError:&error];
if (outputData == nil || error != nil) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
uint8_t output[2U];
[outputData getBytes:output length:(sizeof(uint8_t) * 2U)];
float dequantized[2U];
dequantized[0] = params.scale * (output[0] - params.zeroPoint);
dequantized[1] = params.scale * (output[1] - params.zeroPoint);
[weakSelf updateResultsText:
[NSString stringWithFormat:@"Performing 2 add operations on quantized input:\n\n"
@"Input = [%d, %d]\n\nQuantized Output = [%d, %d]\n\n"
@"Dequantized Output = [%f, %f]",
one, three, output[0], output[1], dequantized[0],
dequantized[1]]];
});
}
- (void)invokeMultiAdd {
__weak typeof(self) weakSelf = self;
dispatch_async(self.interpreterSerialQueue, ^{
if (weakSelf.interpreter == nil) {
[weakSelf updateResultsText:kNilInterpreterError];
return;
}
NSArray<NSNumber *> *shape = @[@2];
NSError *error;
for (int i = 0; i < weakSelf.interpreter.inputTensorCount; ++i) {
if (![weakSelf.interpreter resizeInputTensorAtIndex:i toShape:shape error:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
}
if (![weakSelf.interpreter allocateTensorsWithError:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
NSString *inputs = @"";
for (int i = 0; i < weakSelf.interpreter.inputTensorCount; ++i) {
TFLTensor *inputTensor = [weakSelf.interpreter inputTensorAtIndex:i error:&error];
if (inputTensor == nil || error != nil) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
NSMutableData *inputData = [NSMutableData dataWithCapacity:0];
float input1 = (float)(i + 1);
float input2 = (float)(i + 2);
inputs = [NSString stringWithFormat:@"%@%@[%.1f, %.1f]", inputs,
(inputs.length == 0 ? @"[" : @", "), input1, input2];
[inputData appendBytes:&input1 length:sizeof(float)];
[inputData appendBytes:&input2 length:sizeof(float)];
if (![inputTensor copyData:inputData error:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
}
inputs = [NSString stringWithFormat:@"%@]", inputs];
if (![weakSelf.interpreter invokeWithError:&error]) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
NSString *outputs = @"";
for (int i = 0; i < weakSelf.interpreter.outputTensorCount; ++i) {
TFLTensor *outputTensor = [weakSelf.interpreter outputTensorAtIndex:i error:&error];
if (outputTensor == nil || error != nil) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
NSData *outputData = [outputTensor dataWithError:&error];
if (outputData == nil || error != nil) {
[weakSelf updateResultsText:[NSString stringWithFormat:kInvokeInterpreterError,
error.localizedDescription]];
return;
}
float output[2U];
[outputData getBytes:output length:(sizeof(float) * 2U)];
outputs =
[NSString stringWithFormat:@"%@%@[%.1f, %.1f]", outputs,
(outputs.length == 0 ? @"[" : @", "), output[0], output[1]];
}
outputs = [NSString stringWithFormat:@"%@]", outputs];
[weakSelf
updateResultsText:
[NSString
stringWithFormat:@"Performing 3 add operations:\n\nInputs = %@\n\nOutputs = %@",
inputs, outputs]];
});
}
- (void)updateResultsText:(NSString *)text {
__weak typeof(self) weakSelf = self;
TLTSafeDispatchOnMain(^{
weakSelf.resultsTextView.text = text;
});
}
@end
NS_ASSUME_NONNULL_END

22
tensorflow/lite/experimental/objc/apps/TestApp/TestApp/main.m Normal file
View File

@ -0,0 +1,22 @@
// Copyright 2019 Google Inc. 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.
#import <UIKit/UIKit.h>
#import "AppDelegate.h"
int main(int argc, char* argv[]) {
@autoreleasepool {
return UIApplicationMain(argc, argv, nil, NSStringFromClass([AppDelegate class]));
}
}

19
tensorflow/lite/experimental/objc/tests/TFLInterpreterTests.m
View File

@ -33,9 +33,6 @@ static NSString *const kAddQuantizedModelResourceName = @"add_quantized";
/** Model resource type. */
static NSString *const kAddModelResourceType = @"bin";
/** Rank of the input and output tensor in the Add model. */
static const NSUInteger kAddModelTensorRank = 1U;
/** Size of the first (and only) dimension of the input and output tensor in the Add model. */
static const NSUInteger kAddModelTensorFirstDimensionSize = 2U;
@ -100,8 +97,7 @@ static const float kTestAccuracy = 1E-5F;
- (void)testSuccessfulFullRunAddFloatModel {
// Shape for both input and output tensor.
NSMutableArray *shape = [NSMutableArray arrayWithCapacity:kAddModelTensorRank];
shape[0] = [NSNumber numberWithUnsignedInteger:kAddModelTensorFirstDimensionSize];
NSArray<NSNumber *> *shape = @[@(kAddModelTensorFirstDimensionSize)];
// Creates the interpreter options.
TFLInterpreterOptions *options = [[TFLInterpreterOptions alloc] init];
@ -184,8 +180,7 @@ static const float kTestAccuracy = 1E-5F;
- (void)testSuccessfulFullRunQuantizedModel {
// Shape for both input and output tensor.
NSMutableArray *shape = [NSMutableArray arrayWithCapacity:kAddModelTensorRank];
shape[0] = [NSNumber numberWithUnsignedInteger:kAddModelTensorFirstDimensionSize];
NSArray<NSNumber *> *shape = @[@(kAddModelTensorFirstDimensionSize)];
// Creates the interpreter options.
TFLInterpreterOptions *options = [[TFLInterpreterOptions alloc] init];
@ -276,10 +271,6 @@ static const float kTestAccuracy = 1E-5F;
}
- (void)testInitWithModelPath_invalidPath {
// Shape for both input and output tensor.
NSMutableArray *shape = [NSMutableArray arrayWithCapacity:kAddModelTensorRank];
shape[0] = [NSNumber numberWithUnsignedInteger:kAddModelTensorFirstDimensionSize];
// Creates the interpreter.
NSError *error;
TFLInterpreter *brokenInterpreter = [[TFLInterpreter alloc] initWithModelPath:@"InvalidPath"
@ -308,8 +299,7 @@ static const float kTestAccuracy = 1E-5F;
}
- (void)testResizeInputTensorAtIndex_invalidIndex {
NSMutableArray *shape = [NSMutableArray arrayWithCapacity:kAddModelTensorRank];
shape[0] = [NSNumber numberWithUnsignedInteger:kAddModelTensorFirstDimensionSize];
NSArray<NSNumber *> *shape = @[@(kAddModelTensorFirstDimensionSize)];
NSError *error;
XCTAssertFalse([self.interpreter resizeInputTensorAtIndex:kInvalidInputTensorIndex
toShape:shape
@ -325,8 +315,7 @@ static const float kTestAccuracy = 1E-5F;
}
- (void)testResizeInputTensorAtIndex_zeroDimensionSize {
NSMutableArray *shape = [NSMutableArray arrayWithCapacity:kAddModelTensorRank];
shape[0] = [NSNumber numberWithUnsignedInteger:0];
NSArray<NSNumber *> *shape = @[@0];
NSError *error;
XCTAssertFalse([self.interpreter resizeInputTensorAtIndex:0 toShape:shape error:&error]);
XCTAssertEqual(error.code, TFLInterpreterErrorCodeInvalidShape);

14
tensorflow/lite/experimental/swift/TestApp/TestApp/AppDelegate.swift
View File

@ -1,3 +1,17 @@
// Copyright 2019 Google Inc. 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.
import UIKit
@UIApplicationMain

14
tensorflow/lite/experimental/swift/TestApp/TestApp/Array+TensorFlowLite.swift
View File

@ -1,3 +1,17 @@
// Copyright 2019 Google Inc. 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.
import Foundation
extension Array {

14
tensorflow/lite/experimental/swift/TestApp/TestApp/Data+TensorFlowLite.swift
View File

@ -1,3 +1,17 @@
// Copyright 2019 Google Inc. 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.
import Foundation
extension Data {

14
tensorflow/lite/experimental/swift/TestApp/TestApp/ViewController.swift
View File

@ -1,3 +1,17 @@
// Copyright 2019 Google Inc. 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.
import class TensorFlowLite.Interpreter
import struct TensorFlowLite.InterpreterOptions
import struct TensorFlowLite.Tensor