Sie sind auf Seite 1von 2

//import android.widget.

EditText;
//import android.widget.TextView;
import org.tensorflow.lite.Interpreter;

import java.io.FileInputStream;
import java.io.IOException;
import java.nio.MappedByteBuffer;
import java.nio.channels.FileChannel;

public class Main3Activity extends AppCompatActivity{


//EditText inputNumber;
Button inferButton;
//TextView outputNumber;
Interpreter tflite;

@Override
protected void onCreate(Bundle savedInstanceState){
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main3);
//inputNumber=(EditText)findViewById(R.id.inputNumber)
//outputNumber=(TextView)findViewById(R.id.outputNumber);
inferButton=(Button)findViewById(R.id.inferButton);

try{
tflite=new Interpreter(loadModelFile());
}catch(Exception ex){
ex.printStackTrace();
}
//When we click the infer button, we should do the inference
inferButton.setOnClickListener(new View.OnClickListener(){
@override
public void onClick(View view){
float
prediction=doInference(inputNumber.getText().toString());
outputNumber.setText(Float.toSting(prediction));
}
});
}
public float doInference(String inputString){
//Input shape is [1]. Single valued input
float[] inputVal=new float[1];
inputVal[0]=Float.valueOf(inputString);

//Output shape is [1][1]


float[][] outputval=new float[1][1];

//Run inference passing the input shape and getting the output shape
tflite.run(inputVal, outputVal);

//Inferred value is at [0][0]


float inferredValue=outputValue[0][0];

return inferredValue;
}

//Memory-map the model file in Assets


private MappedByteBuffer loadModelFile() throws IOException{
//Open the model using an input stream, and memory map it to load
AssetFileDescriptor fileDescriptor=this.getAssets().openFd(**)
FileInputStream inputStream=new
FileInputStream(fileDescriptor.getFileDescriptor());
long startOffset=fileDescriptor.getStartOffset();
long declaredLength=fileDescriptor.getDeclaredLength();
return fileChannel.map(FileChannel.MapMode.READ_ONLY, startOffset,
declaredLength);
}
}
<activity android:name=".MainActivity">
<intent-filter>
<action android:name="android.intent.action.MAIN" />

<category android:name="android.intent.category.LAUNCHER" />


</intent-filter>

Das könnte Ihnen auch gefallen