tflite

README

MLModelScope Tensorflow Lite Agent

This is the Tensorflow Lite agent for MLModelScope, an open-source framework and hardware agnostic, extensible and customizable platform for evaluating and profiling ML models across datasets / frameworks / systems, and within AI application pipelines. It is used to deploy mobile devices through MLModelScope ecosystem.

Currently it has most of the models from Tensorflow Lite Model Zoo built in, plus many models acquired from public repositories. Although the agent supports different modalities including Object Detection and Image Enhancement, most of the built-in models are for Image Classification. More built-in models are coming. One can evaluate the ~50 models on any mobile device of interest as long as MLModelScope Mobile Agent is deployed on it.

Check out MLModelScope and welcome to contribute.

Note: rest of the documentation to be added.

Documentation

Index

• Variables
• func Asset(name string) ([]byte, error)
• func AssetDir(name string) ([]string, error)
• func AssetInfo(name string) (os.FileInfo, error)
• func AssetNames() []string
• func MustAsset(name string) []byte
• func Register()
• func RestoreAsset(dir, name string) error
• func RestoreAssets(dir, name string) error

Variables

var (
	Version   = "0.1.0"
	BuildDate = "undefined"
	GitCommit = "undefined"
)

Version ...

var FrameworkManifest = dlframework.FrameworkManifest{
	Name:    "TensorflowLite",
	Version: "1.15",
	Container: map[string]*dlframework.ContainerHardware{
		"amd64": {
			Cpu: "raiproject/carml-pytorch:amd64-cpu",
			Gpu: "raiproject/carml-pytorch:amd64-gpu",
		},
		"ppc64le": {
			Cpu: "raiproject/carml-pytorch:ppc64le-gpu",
			Gpu: "raiproject/carml-pytorch:ppc64le-gpu",
		},
	},
}

Functions

func Asset

func Asset(name string) ([]byte, error)

Asset loads and returns the asset for the given name. It returns an error if the asset could not be found or could not be loaded.

func AssetDir

func AssetDir(name string) ([]string, error)

AssetDir returns the file names below a certain directory embedded in the file by go-bindata. For example if you run go-bindata on data/... and data contains the following hierarchy:

data/
  foo.txt
  img/
    a.png
    b.png

then AssetDir("data") would return []string{"foo.txt", "img"} AssetDir("data/img") would return []string{"a.png", "b.png"} AssetDir("foo.txt") and AssetDir("notexist") would return an error AssetDir("") will return []string{"data"}.

func AssetInfo

func AssetInfo(name string) (os.FileInfo, error)

AssetInfo loads and returns the asset info for the given name. It returns an error if the asset could not be found or could not be loaded.

func AssetNames

func AssetNames() []string

AssetNames returns the names of the assets.

func MustAsset

func MustAsset(name string) []byte

MustAsset is like Asset but panics when Asset would return an error. It simplifies safe initialization of global variables.

func Register

func Register()

func RestoreAsset

func RestoreAsset(dir, name string) error

RestoreAsset restores an asset under the given directory

func RestoreAssets

func RestoreAssets(dir, name string) error

RestoreAssets restores an asset under the given directory recursively