Advertisement
Guest User

Untitled

a guest
Jul 25th, 2017
55
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.67 KB | None | 0 0
  1. ...
  2. public class MainActivity extends AppCompatActivity {
  3.  
  4. ...
  5. private TensorFlowInferenceInterface mInferenceInterface;
  6.  
  7. @Override
  8. protected void onCreate(Bundle savedInstanceState) {
  9. super.onCreate(savedInstanceState);
  10. setContentView(R.layout.activity_main);
  11.  
  12. mInferenceInterface = new TensorFlowInferenceInterface(getAssets(), "file:///android_asset/optimized_tfdroid.pb");
  13.  
  14. ...
  15. float[] inputFloats = {1, 1, 1};
  16. mInferenceInterface.feed("I", inputFloats, 1, 3);
  17. mInferenceInterface.run(new String[]{"O"});
  18. float[] resu = {0, 0};
  19. mInferenceInterface.fetch("O", resu);
  20. ...
  21. }
  22. ...
  23. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement