/caffe-android-lib

Porting caffe to android platform

Primary LanguageC++OtherNOASSERTION

Caffe-Android-Lib

Goal

Porting caffe to android platform

Build

git clone --recursive https://github.com/sh1r0/caffe-android-lib.git
cd caffe-android-lib
./build.py $(NDK_PATH)

Usage

  • put required stuff into your device

     ./get_model.py
 adb shell mkdir -p /sdcard/caffe_mobile/
 adb push caffe-mobile/jni/caffe/models/bvlc_reference_caffenet/ /sdcard/caffe_mobile/bvlc_reference_caffenet/

  • copy caffe-mobile/libs/armeabi-v7a/*.so to your jni lib directory

  • in your main activity

     static {
 	System.loadLibrary("caffe");
 	System.loadLibrary("caffe_jni");
 }

  • create CaffeMobile.java

     package com.sh1r0.caffe_android_demo;

 public class CaffeMobile {
 	public native void enableLog(boolean enabled);
 	public native int loadModel(String modelPath, String weightsPath);
 	public native int predictImage(String imgPath);
 }

  • call native methods

     CaffeMobile caffeMobile = new CaffeMobile();
 caffeMobile.enableLog(true);  // optional, enable native logging
 caffeMobile.loadModel(modelPath, weightsPath);  // init once
 ...
 caffeMobile.predictImage(imgPath);

Optional

.envrc files are for direnv

direnv is an environment variable manager for your shell. It knows how to hook into bash, zsh and fish shell to load or unload environment variables depending on your current directory. This allows to have project-specific environment variables and not clutter the "~/.profile" file.

Dependency

Credits