Skip to content

Instantly share code, notes, and snippets.

@frogermcs
Created June 9, 2019 13:07
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
Star You must be signed in to star a gist
Save frogermcs/65187f00b8ef6b5ebf602d568af2a5db to your computer and use it in GitHub Desktop.
public class ModelTestActivity extends AppCompatActivity {
private ImageView ivPreview;
private TextView tvClassification;
private ModelClassificator modelClassificator;
@Override
protected void onCreate(@Nullable Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(com.frogermcs.imageclassificationtester.test.R.layout.activity_model_test);
ivPreview = findViewById(com.frogermcs.imageclassificationtester.test.R.id.ivPreview);
tvClassification = findViewById(com.frogermcs.imageclassificationtester.test.R.id.tvClassification);
initClassification();
}
private void initClassification() {
try {
ModelConfig modelConfig = new FlowersConfig();
modelClassificator = new ModelClassificator(this, modelConfig);
} catch (IOException e) {
e.printStackTrace();
Toast.makeText(this, "Frame Processor initialization failed", Toast.LENGTH_SHORT).show();
}
}
public void classifyImage(Bitmap bitmap) {
setImagePreview(bitmap);
performClassification(bitmap);
}
public void setImagePreview(Bitmap bitmap) {
Bitmap squareBitmap = ThumbnailUtils.extractThumbnail(bitmap, 100, 100);
runOnUiThread(() -> setImage(squareBitmap));
}
private void performClassification(Bitmap bitmap) {
List<ClassificationResult> classificationResults = modelClassificator.process(bitmap);
showClassificationResults(classificationResults);
}
public void showClassificationResults(List<ClassificationResult> classificationResults) {
runOnUiThread(() -> tvClassification.setText(ResultsUtils.resultsToStr(classificationResults)));
}
public void setImage(Bitmap bmp) {
ivPreview.setImageBitmap(bmp);
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment