kurtjacobsdev / torch2ios

Torch7 Library - Convert NN Models To iOS Format

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

torch2ios

Small lib to serialise Torch7 Networks for iOS. Supported Layers include Fully Connected, Pooling and Convolution Layers at present. The library stores the weights & biases (if any) for each layer necesarry for inference on iOS devices.

Usage is simple: just require 'torch2ios' in your lua script and call saveForiOS(_yourmodel, _destfilename) and a new .t7ios file will be created for use with the KSJNeuralNetwork [https://github.com/woffle/KSJNeuralNetwork] iOS library on the iOS client.

Included is a simple demo of the usage, see xor.lua.

Also added are API to access the binary file format on iOS and C. See iOSClient for iOS example and torch2ios_read.c for C example. Namespace for the iOS to Torch7 support will follow THES (Torch Embedded Systems)tag.

alt tag alt tag alt tag alt tag

About

Torch7 Library - Convert NN Models To iOS Format


Languages

Language:Objective-C 49.3%Language:Lua 29.4%Language:C 21.3%