Model

Model kelas publik

Kelas pembungkus untuk model TFLite dan penerjemah TFLite.

Catatan: Suatu Model hanya dapat menampung 1 model TFLite dalam satu waktu, dan selalu menampung instance interpreter TFLite untuk menjalankannya.

Kelas Bersarang

kelas Model.Pembangun Kelas ini tidak digunakan lagi. Silakan gunakan Model.createModel(Context, String, Options) .
enum Model.Perangkat Jenis perangkat runtime yang digunakan untuk menjalankan klasifikasi.
kelas Model.Pilihan Opsi untuk menjalankan model.

Metode Publik

ruang kosong
menutup ()
Model statis
createModel (Konteks konteks, String modelPath, opsi Model.Options )
Memuat model dari aset dan menginisialisasi juru bahasa TFLite dengan opsi yang diberikan.
Model statis
createModel (Konteks konteks, String modelPath)
Memuat model dari aset dan menginisialisasi juru bahasa TFLite.
MappedByteBuffer
dapatkanData ()
Mengembalikan data model yang dipetakan memori.
Tensor
getInputTensor (int inputIndex)
Mengasosiasikan Tensor dengan indeks masukan yang disediakan.
Tensor
getOutputTensor (int keluaranIndeks)
Mengasosiasikan Tensor dengan indeks keluaran yang disediakan.
ke dalam[]
getOutputTensorShape (int keluaranIndeks)
Mengembalikan bentuk keluaran.
Rangkaian
dapatkan Jalur ()
Mengembalikan jalur file model yang disimpan di Aset.
ruang kosong
jalankan (masukan Objek[] , keluaran Peta < Integer , Objek >)
Menjalankan inferensi model pada beberapa masukan, dan mengembalikan beberapa keluaran.

Metode Warisan

Metode Publik

penutupan kekosongan publik ()

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

Memuat model dari aset dan menginisialisasi juru bahasa TFLite dengan opsi yang diberikan.

Parameter
konteks Konteks Aplikasi.
modelPath Jalur file model.
pilihan Opsi untuk menjalankan model.
Melempar
Pengecualian IO jika ada pengecualian saat membuka file model.
Lihat juga

Model statis publik createModel (Konteks konteks, String modelPath)

Memuat model dari aset dan menginisialisasi juru bahasa TFLite.

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

Parameter
konteks Konteks Aplikasi.
modelPath Jalur file model.
Melempar
Pengecualian IO jika ada pengecualian saat membuka file model.

publik MappedByteBuffer getData ()

Mengembalikan data model yang dipetakan memori.

Tensor publik getInputTensor (int inputIndex)

Mengasosiasikan Tensor dengan indeks masukan yang disediakan.

Parameter
masukanIndeks
Melempar
IlegalStateException jika penerjemahnya tertutup.

Tensor publik getOutputTensor (int outputIndex)

Mengasosiasikan Tensor dengan indeks keluaran yang disediakan.

Parameter
Indeks keluaran
Melempar
IlegalStateException jika penerjemahnya tertutup.

int publik[] getOutputTensorShape (int outputIndex)

Mengembalikan bentuk keluaran. Berguna jika bentuk keluaran hanya ditentukan saat grafik dibuat.

Parameter
Indeks keluaran
Melempar
IlegalStateException jika penerjemahnya tertutup.

GetPath String publik ()

Mengembalikan jalur file model yang disimpan di Aset.

public void run (masukan Object[] , keluaran Map < Integer , Object >)

Menjalankan inferensi model pada beberapa masukan, dan mengembalikan beberapa keluaran.

Parameter
masukan serangkaian data masukan. Inputnya harus dalam urutan yang sama dengan input model. Setiap input dapat berupa array atau array multidimensi, atau ByteBuffer dengan tipe primitif termasuk int, float, long, dan byte. ByteBuffer adalah cara yang lebih disukai untuk meneruskan data masukan berukuran besar, sedangkan tipe string memerlukan penggunaan jalur masukan array (multidimensi). Saat ByteBuffer digunakan, kontennya tidak boleh berubah hingga inferensi model selesai.
keluaran peta yang memetakan indeks keluaran ke array multidimensi data keluaran atau ByteBuffer tipe primitif termasuk int, float, long, dan byte. Ia hanya perlu menyimpan entri agar output dapat digunakan.