Model

class public Model

Kelas wrapper untuk model TFLite dan penafsir TFLite.

Catatan: Model hanya dapat menyimpan 1 model TFLite dalam satu waktu, dan selalu memiliki instance penafsir TFLite untuk menjalankannya.

Class Bertingkat

class Model.Builder Class ini tidak digunakan lagi. Harap gunakan Model.createModel(Context, String, Options).  
enum Model.Device Jenis perangkat runtime yang digunakan untuk menjalankan klasifikasi. 
class Model.Options Opsi untuk menjalankan model. 

Metode Publik

void
close()
Model statis
createModel(Konteks konteks, opsi String modelPath, Model.Options)
Memuat model dari aset dan melakukan inisialisasi penafsir TFLite dengan opsi yang diberikan.
Model statis
createModel(Konteks konteks, String modelPath)
Memuat model dari aset dan melakukan inisialisasi penafsir TFLite.
MappedByteBuffer
getData()
Menampilkan data model yang dipetakan memori.
Tensor
getInputTensor(int inputIndex)
Mendapatkan Tensor yang terkait dengan indeks input yang disediakan.
Tensor
getOutputTensor(int outputIndex)
Mendapatkan Tensor yang terkait dengan indeks output yang disediakan.
int[]
getOutputTensorShape(int outputIndex)
Menampilkan bentuk output.
String
getPath()
Menampilkan jalur file model yang disimpan di Aset.
void
run(input Object[], output Map<Integer, Object> output)
Menjalankan inferensi model pada beberapa input, dan menampilkan beberapa output.

Metode Turunan

Metode Publik

public void close ()

public static Model createModel (Konteks konteks, String modelPath, Model.Options opsi)

Memuat model dari aset dan melakukan inisialisasi penafsir TFLite dengan opsi yang diberikan.

Parameter
context Konteks Aplikasi.
modelPath Jalur file model.
opsi Opsi untuk menjalankan model.
Menampilkan
IOException jika ada pengecualian yang terjadi ketika membuka file model.
Lihat Juga

publik statis Model createModel (Konteks konteks, String modelPath)

Memuat model dari aset dan melakukan inisialisasi penafsir TFLite.

Opsi defaultnya adalah: (1) perangkat CPU; (2) satu thread.

Parameter
context Konteks Aplikasi.
modelPath Jalur file model.
Menampilkan
IOException jika ada pengecualian yang terjadi ketika membuka file model.

publik MappedByteBuffer getData ()

Menampilkan data model yang dipetakan memori.

public Tensor getInputTensor (int inputIndex)

Mendapatkan Tensor yang terkait dengan indeks input yang disediakan.

Parameter
inputIndex
Menampilkan
IllegalStateException jika penafsir ditutup.

public Tensor getOutputTensor (int outputIndex)

Mendapatkan Tensor yang terkait dengan indeks output yang disediakan.

Parameter
outputIndex
Menampilkan
IllegalStateException jika penafsir ditutup.

public int[] getOutputTensorShape (int outputIndex)

Menampilkan bentuk output. Berguna jika bentuk output hanya ditentukan saat grafik dibuat.

Parameter
outputIndex
Menampilkan
IllegalStateException jika penafsir ditutup.

public String getPath ()

Menampilkan jalur file model yang disimpan di Aset.

public void run (Object[] input, Map<Integer, Object> output)

Menjalankan inferensi model pada beberapa input, dan menampilkan beberapa output.

Parameter
input array data input. Input harus dalam urutan yang sama dengan input model. Setiap input dapat berupa array atau array multidimensi, atau ByteBuffer jenis elemen primitif seperti int, float, long, dan byte. ByteBuffer adalah cara yang direkomendasikan untuk meneruskan data input besar, sedangkan jenis string memerlukan penggunaan jalur input array (multidimensi). Saat ByteBuffer digunakan, kontennya tidak boleh berubah sampai inferensi model selesai.
output peta yang memetakan output indeks ke array multidimensi dari data output atau ByteBuffer dari jenis primitif seperti int, float, long, dan byte. Modul ini hanya perlu menyimpan entri agar output yang akan digunakan.