Bender基于Metal的机器学习框架可在iOS上运行TensorFlow模型

Bender基于Metal的机器学习框架可在iOS上运行TensorFlow模型

Bender

Bender is an abstraction layer over MetalPerformanceShaders useful for working with neural networks.

Contents

IntroductionWhy did we need BenderBasic usageRequirementsGetting involvedExamplesInstallationChangelog

The documentation can be found under the Documentation folder:

API contains the most important information to get started.Supported Layers explains which layers are supported and how they map to TensorFlow ops.Importing explains how to import models from other frameworks such as TensorFlow. You can also find information on how to enhance this functionality for custom implementations.

Introduction

Bender is an abstraction layer over MetalPerformanceShaders which is used to work with neural networks. It is of growing interest in the AI environment to execute neural networks on mobile devices even if the training process has been done previously. We want to make it easier for everyone to execute pretrained networks on iOS.

Bender allows you to easily define and run neural networks using the most common layers like Convolution, Pooling, FullyConnected and some normalizations among others. It is also flexible in the way it receives the parameters for these layers.

We also want to support loading models trained on other frameworks such as TensorFlow or Caffe2. Currently Bender includes an adapter for TensorFlow that loads a graph with variables and "translates" it to Bender layers. This feature supports a subset of TensorFlow's operations but we plan to enhance it to cover more cases.

Why did we need Bender?

At Xmartlabs we were about to start a Machine Learning project and investigated frameworks to use in iOS. We found MetalPerformanceShaders useful but not very user friendly and we saw ourselves repeating a lot of code and information. That is why we starting building a framework to handle that kind of stuff.

We also found ourselves creating scripts to translate the models we had from training with TensorFlow to iOS. This means transposing the weights to the MPSCNN format and also mapping the parameters of the different kinds of layers in TensorFlow to the parameters used by the MPSCNN kernels. TensorFlow can be compiled for iOS but currently it does not support running on GPU which we wanted to do. We also did not want to include TensorFlow's static library into our project. This is why we also started to work on an adapter that would parse a TF graph and translate it to our Bender layers.

Usage

You can define your own network in Bender using our custom operator or you can load a model exported from TensorFlow. Defining a network and loading a model can be done like this:

import MetalBenderlet url = Bundle.main.url(forResource: "myModel", withExtension: "pb")! // A TensorFlow model.let network = Network.load(url: url, inputSize: LayerSize(h: 256, w: 256, f: 3))network.run(input: /* ... */) { output in // ...}

You can read more information about this in Importing.

If you want to define your network yourself you can do it like this:

let network = Network(inputSize: LayerSize(h: 256, w: 256, f: 3))network.start ->> Convolution(convSize: ConvSize(outputChannels: 16, kernelSize: 3, stride: 2)) ->> InstanceNorm() ->> Convolution(convSize: ConvSize(outputChannels: 32, kernelSize: 3, stride: 2), neuronType: .relu) ->> InstanceNorm() ->> FullyConnected(neurons: 128) ->> Neuron(type: .tanh) ->> FullyConnected(neurons: 10) ->> Softmax()// ...

and once you're done with all your layers:

network.initialize()

To know more about this have a look at API.

Requirements

Xcode 9iOS 11.0+ (but deployment target is iOS 10.0, so iOS 10 is supported)

Getting involved

If you want to contribute please feel free to submit pull requests.If you have a feature request please open an issue.If you found a bug or need help please check older issues, FAQ and threads on StackOverflow before submitting an issue.

Before contribute check the CONTRIBUTING file for more info.

If you use Bender in your app We would love to hear about it! Drop us a line on Twitter.

Examples

Follow these steps to run the examples:

Clone Bender repository (or download it).Run carthage update --platform iOS in the downloaded folder.Open Bender workspace and run the Example project.

There is an Image recognition example which includes a MobileNet model in Bender and one in CoreML. It is also set up to run an Inception model but you will have to download it separately as it is almost 100 MB in size. You can download it from http://download.tensorflow.org/models/inception_v3_2016_08_28.tar.gz but then you have to freeze it and add it to the 'Example' Xcode project as 'inception_v3.pb'.

Installation

CocoaPods

To install Bender, simply add the following line to your Podfile:

pod 'MetalBender', '~> 0.5'

Remember that Bender compiles for iOS 10. So you must add platform :ios, '10.0' to your Podfile

Carthage

Carthage is a simple, decentralized dependency manager for Cocoa.

To install Bender, add the following line to your Cartfile:

github "xmartlabs/Bender" ~> 0.5

Then run:

carthage update --platform iOS

Finally, drag the built .framework binaries for MetalBender, MetalPerformanceShadersProxy and SwiftProtobuf to your application's Xcode project.

Author

Xmartlabs SRL (@xmartlabs)

Change Log

This can be found in the CHANGELOG.md file.

License

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。