Как передать изображение в предварительно обученную модель TFLite в android, используя Java? - PullRequest
Новая
       25

Как передать изображение в предварительно обученную модель TFLite в android, используя Java?

0 голосов
/ 19 апреля 2020

Я новичок в android разработке, мне нужно получить предварительно обученную модель классификации изображений и запустить ее с выбранным изображением, мне удалось выбрать изображения из галереи и извлечь изображение из URI, кроме загрузки TFLite модель

Я также видел учебники о преобразованиях модели TFLite и смог запустить простую модель преобразования C градусов в F градусов

Я не могу сделать это с изображением, какая-нибудь помощь?

Android код для выбора изображения и загрузки модели: 

import android.annotation.SuppressLint;
import android.app.Activity;
import android.content.Intent;
import android.content.res.AssetFileDescriptor;
import android.net.Uri;
import android.os.Bundle;
import android.view.View;
import android.widget.Button;
import android.widget.ImageView;
import android.widget.TextView;

import androidx.appcompat.app.AppCompatActivity;

import org.tensorflow.lite.Interpreter;

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

public class MainActivity extends AppCompatActivity {

    Button btnChoose;
    ImageView imageView;
    Interpreter tflite;
    TextView textView;

    String[] classes;

    private static int GALLERY_REQUEST_CODE = 35;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);

        imageView = findViewById(R.id.imageView);
        textView = findViewById(R.id.textView);

        btnChoose = findViewById(R.id.btnChoose);

        btnChoose.setOnClickListener(new View.OnClickListener() {
            @SuppressLint("IntentReset")
            @Override
            public void onClick(View v) {
                Intent intent=new Intent(Intent.ACTION_PICK);
                intent.setType("image/*");
                String[] mimeTypes = {"image/jpeg", "image/png"};
                intent.putExtra(Intent.EXTRA_MIME_TYPES,mimeTypes);
                startActivityForResult(intent,GALLERY_REQUEST_CODE);
            }
        });

        try{
            tflite = new Interpreter(loadModelFile());
        } catch (Exception ex){
            ex.printStackTrace();
        }

        try{
            InputStream is = getAssets().open("labels_mobilenet_quant_v1_224.txt");
            int size = is.available();
            byte[] buffer = new byte[size];
            is.read(buffer);
            is.close();
            classes = new String(buffer).split("\n");
            textView.setText(classes[1]);
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }


    @Override
    protected void onActivityResult(int requestCode, int resultCode, Intent data) {
        super.onActivityResult(requestCode, resultCode, data);

        if (requestCode == GALLERY_REQUEST_CODE && resultCode == Activity.RESULT_OK && null != data) {
            Uri selectedImage = data.getData();
            imageView.setImageURI(selectedImage);
        }
    }

    // TODO:
    /*
    doInf Function
     */

    private MappedByteBuffer loadModelFile() throws IOException {
        AssetFileDescriptor fileDescriptor = this.getAssets().openFd("mobilenet_v1_1.0_224_quant.tflite");
        FileInputStream inputStream = new FileInputStream(fileDescriptor.getFileDescriptor());
        FileChannel fileChannel = inputStream.getChannel();
        long startOffset = fileDescriptor.getStartOffset();
        long declaredLength = fileDescriptor.getDeclaredLength();
        return fileChannel.map(FileChannel.MapMode.READ_ONLY, startOffset, declaredLength);
    }

}

TFLite Простой код модели, который я сделал до 

Button btn;
TextView tvOutput;
EditText etInput;
Interpreter tflite;

@Override
protected void onCreate(Bundle savedInstanceState) {
    super.onCreate(savedInstanceState);
    setContentView(R.layout.activity_main);

    btn = findViewById(R.id.button);
    etInput = findViewById(R.id.etInput);
    tvOutput = findViewById(R.id.tvOutput);

    try{
        tflite = new Interpreter(loadModelFile());
    } catch (Exception ex){
        ex.printStackTrace();
    }

    btn.setOnClickListener(new View.OnClickListener() {
        @SuppressLint("SetTextI18n")
        @Override
        public void onClick(View v) {
            float pred = doInf(etInput.getText().toString());
            tvOutput.setText(Float.toString(pred));
        }
    });
}

public float doInf(String inputString){
    float[] inputVal = new float[1];
    inputVal[0] = Float.parseFloat(inputString);

    float[][] outputVal = new float[1][1];

    tflite.run(inputVal, outputVal);

    return outputVal[0][0];
}

private MappedByteBuffer loadModelFile() throws IOException{
    AssetFileDescriptor fileDescriptor = this.getAssets().openFd("linear.tflite");
    FileInputStream inputStream = new FileInputStream(fileDescriptor.getFileDescriptor());
    FileChannel fileChannel = inputStream.getChannel();
    long startOffset = fileDescriptor.getStartOffset();
    long declaredLength = fileDescriptor.getDeclaredLength();
    return fileChannel.map(FileChannel.MapMode.READ_ONLY, startOffset, declaredLength);
}
...