in ImageSegmentation/app/src/main/java/org/pytorch/imagesegmentation/MainActivity.java [64:115]
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
try {
mBitmap = BitmapFactory.decodeStream(getAssets().open(mImagename));
} catch (IOException e) {
Log.e("ImageSegmentation", "Error reading assets", e);
finish();
}
mImageView = findViewById(R.id.imageView);
mImageView.setImageBitmap(mBitmap);
final Button buttonRestart = findViewById(R.id.restartButton);
buttonRestart.setOnClickListener(new View.OnClickListener() {
public void onClick(View v) {
if (mImagename == "deeplab.jpg")
mImagename = "dog.jpg";
else
mImagename = "deeplab.jpg";
try {
mBitmap = BitmapFactory.decodeStream(getAssets().open(mImagename));
mImageView.setImageBitmap(mBitmap);
} catch (IOException e) {
Log.e("ImageSegmentation", "Error reading assets", e);
finish();
}
}
});
mButtonSegment = findViewById(R.id.segmentButton);
mProgressBar = (ProgressBar) findViewById(R.id.progressBar);
mButtonSegment.setOnClickListener(new View.OnClickListener() {
public void onClick(View v) {
mButtonSegment.setEnabled(false);
mProgressBar.setVisibility(ProgressBar.VISIBLE);
mButtonSegment.setText(getString(R.string.run_model));
Thread thread = new Thread(MainActivity.this);
thread.start();
}
});
try {
mModule = LiteModuleLoader.load(MainActivity.assetFilePath(getApplicationContext(), "deeplabv3_scripted_optimized.ptl"));
} catch (IOException e) {
Log.e("ImageSegmentation", "Error reading assets", e);
finish();
}
}