ClassificationFrameProcessor.java

public class ClassificationFrameProcessor implements FrameProcessor {

    private final Interpreter interpreter;
    private final List<String> labels;
    private final ClassificationListener classificationListener;
    private final ModelConfig modelConfig;

    /* ... */

    @Override
    public void process(@NonNull Frame frame) {
        Bitmap bitmap = ImageUtils.frameToBitmap(frame);
        Bitmap toClassify = ThumbnailUtils.extractThumbnail(
                bitmap, modelConfig.getInputWidth(), modelConfig.getInputHeight()
        );
        bitmap.recycle();

        ByteBuffer byteBufferToClassify = bitmapToModelsMatchingByteBuffer(toClassify);
        float[][] result = new float[1][labels.size()];
        interpreter.run(byteBufferToClassify, result);
        classificationListener.onClassifiedFrame(getSortedResult(result));
    }

    /* ... */

    public interface ClassificationListener {
        void onClassifiedFrame(List<ClassificationResult> classificationResults);
    }
}