Neural style in Android
Arcade is an experimental Android port of Torch-7 implementation of neural-style.
Get the demo app from Google Play
This repository contains prebuilt shared libraries needed to build torch-android and libarcade.so. If you want to build the shared libraries go to neural-style-android which is based on top of torch-android and adds support for Protobuf and loadcaffe. NIN ImageNet models are used in Arcade due to smaller size than VVG models. Models have to be seperately downloaded. Demo app module have a ModelDownloader class to download and place models in correct path
Build the libraries by NDK
ndk-build and then directly run from Android studio. The build configuration is taken from
src/main/jni and Gradle's native build system is ignored (Gradle currently ignores existing Android.mk and I was unable to figure out how to include prebuilt shared libraries from Gradle).
Note - only
armeabi-v7a libraries are built currently and app will not work on other architectures.
Arcade is built as a seperate Android library and contains a Builder for all styling settings and helper functions. You can use this library to create your own implementation. Regular callbacks are also provided from Lua -> C -> Java for progress, iteration updates, completion and Images saved listeners.
Most of the middelware code between java and lua is located in arcade.cpp
Usage is pretty straightforward. Compile library project and use builder to setup configuration.
ArcadeBuilder builder = new ArcadeBuilder(this); builder.setStyleimage(stylePath); builder.setContentImage(contentPath); builder.setModelFile(modelPath); builder.setProtoFIle(protoPath); builder.setImageSize(512); builder.setIterations(30); .... Arcade arcade = builder.build(); //initialize and load lua script arcade.initialize(); //set listeners arcade.setProgressListener(progressListener); //begin styling arcade.stylize();
Due to no no GPU and limited processing power and memory, the styling is pretty slow and unusable for image sizes greater than 512. Due to speed limitations, getting respectable result is only unlikely. Better results can be achieved by trying different combination of style settings like style weight, content weight and number of iterations.
30 iterations,Image Size - 256, Device - Nexus 6, Time taken - 25 minutes
15 iterations,Image Size - 512, Device - Nexus 6, Time taken - 40 minutes
I started this project just for experimenting the end result of this thing. I wouldn't say that results are great though can be better by trying different combinations and improving things in the code. On CUDA enabled devices (Tegra K1), this should be a lot faster but support for cutorch is not there currently. Contributions are welcome!