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 |
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. |
---|
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. |
---|
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. |