47 lines
1.5 KiB
Java
47 lines
1.5 KiB
Java
package io.interactionlab.capimgdemo.demo;
|
|
|
|
import android.graphics.Color;
|
|
|
|
/**
|
|
* Created by Huy on 28/06/2018.
|
|
*/
|
|
|
|
public class DemoSettings {
|
|
|
|
private static int[] colorBand(int size) {
|
|
int[] band = new int[size];
|
|
for (int i = 0; i < size; i++) {
|
|
band[i] = Color.HSVToColor(new float[]{i*360.0f/(float)size, 1.0f, 1.0f});
|
|
}
|
|
return band;
|
|
}
|
|
|
|
public static final ModelDescription[] models = new ModelDescription[]{
|
|
new ModelDescription(
|
|
"KnuckleFinger",
|
|
"file:///android_asset/CNN.pb",
|
|
"conv2d_1_input",
|
|
"output_node0",
|
|
new long[]{1, 27, 15, 1},
|
|
new String[]{"Knuckle", "Finger"},
|
|
new int[]{Color.GREEN, Color.YELLOW}
|
|
),
|
|
new ModelDescription(
|
|
"GestureRecognition",
|
|
"file:///android_asset/LSTM.pb",
|
|
"time_distributed_10_input",
|
|
"output_node0",
|
|
new long[]{50, 27, 15, 1},
|
|
new String[]{"Tap", "Two Tap", "Swipe left",
|
|
"Swipe right", "Swipe up", "Swipe down",
|
|
"Swipe up with two", "Swipe down with two", "Circle",
|
|
"Arrowhead left", "Arrowhead right", "Checkmark",
|
|
"Γ", "L", "Mirrored L", "S",
|
|
"Rotate"},
|
|
colorBand(17)
|
|
)
|
|
};
|
|
}
|
|
|
|
|