מחלקת wrapper למודל TFLite ולמתורגמן ב-TFLite.
הערה: Model
יכול להכיל רק מודל TFLite אחד בכל פעם, ותמיד יש בו מכונה של תרגום TFLite כדי להפעיל אותה.
מחלקות מקננות
מחלקה | Model.Builder |
הכיתה הזו הוצאה משימוש. יש להשתמש ב-Model.createModel(Context, String, Options) .
|
|
טיפוסים בני מנייה (enum) | Model.Device | סוג המכשיר בזמן ריצה שמשמש לביצוע הסיווג. | |
מחלקה | Model.Options | אפשרויות להפעלת המודל. |
שיטות ציבוריות
void |
close()
|
סטטי Model |
createModel(הקשר הקשר, מודל String modelPath, אפשרויות Model.Options)
לטעינת מודל מנכסים דיגיטליים ולהפעלה של רכיב התרגום ב-TFLite עם האפשרויות הנתונות.
|
סטטי Model |
createModel(הקשר הקשר, modelPath מחרוזת)
לטעינת מודל מנכסים דיגיטליים ולהפעלה של רכיב התרגום ב-TFLite.
|
MappedByteBuffer |
getData()
מחזירה את נתוני המודל שמופו בזיכרון.
|
טנסור |
getInputTensor(אינדקס קלט)
מקבלת את ה-Tensor שמשויך לאינדקס הקלט הנתון.
|
טנסור |
getOutputTensor(int outputIndex)
מקבלת את Tensor שמשויך לאינדקס הפלט שסופק.
|
int[] |
getOutputTensorShape(אינדקס פלט ב-int)
מחזירה את צורת הפלט.
|
String |
getPath()
מחזירה את הנתיב של קובץ המודל ששמור בקטע 'נכסים'.
|
void |
שיטות שעברו בירושה
שיטות ציבוריות
public void סגירה ()
ציבורי סטטי Model createModel (הקשר הקשר, מחרוזת modelPath, Model.Options)
לטעינת מודל מנכסים דיגיטליים ולהפעלה של רכיב התרגום ב-TFLite עם האפשרויות הנתונות.
פרמטרים
context | ההקשר של האפליקציה |
---|---|
modelPath | הנתיב של קובץ המודל. |
אפשרויות | האפשרויות להפעלת המודל. |
קליעות
IOException | אם מתרחשת חריגה בעת פתיחת קובץ המודל. |
---|
ראו בנוסף
ציבורי סטטי Model createModel (הקשר הקשר, מחרוזת modelPath)
לטעינת מודל מנכסים דיגיטליים ולהפעלה של רכיב התרגום ב-TFLite.
אפשרויות ברירת המחדל הן: (1) מכשיר מעבד (CPU); (2) שרשור אחד.
פרמטרים
context | ההקשר של האפליקציה |
---|---|
modelPath | הנתיב של קובץ המודל. |
קליעות
IOException | אם מתרחשת חריגה בעת פתיחת קובץ המודל. |
---|
Public Tensor getInputTensor (אינדקס קלט פנימי)
מקבלת את ה-Tensor שמשויך לאינדקס הקלט הנתון.
פרמטרים
inputIndex |
---|
קליעות
IllegalStateException | אם המתרגם סגור. |
---|
Public Tensor getOutputTensor (אינדקס פלט פנימי)
מקבלת את Tensor שמשויך לאינדקס הפלט שסופק.
פרמטרים
outputIndex |
---|
קליעות
IllegalStateException | אם המתרגם סגור. |
---|
Public int[] getOutputTensorShape (intoutputIndex)
מחזירה את צורת הפלט. כדאי להשתמש כאן אם צורת הפלט נקבעת רק כשנוצר תרשים.
פרמטרים
outputIndex |
---|
קליעות
IllegalStateException | אם המתרגם סגור. |
---|
Public בטל run (קלט שלאובייקט[], מפה<מספר שלם, פלט> פלט)
מריצה מסקנות מודל על מספר קלטים, ומחזירה מספר פלטים.
פרמטרים
קלט | מערך של נתוני קלט. ערכי הקלט צריכים להיות באותו סדר של ערכי הקלט
במודל. כל קלט יכול להיות מערך או מערך רב ממדי, או ByteBuffer של טיפוסים פרימיטיביים, כולל int, float, ארוך ובייט. ByteBuffer היא הדרך המועדפת להעביר נתוני קלט גדולים, ואילו סוגי מחרוזות
מחייבים שימוש בנתיב הקלט של המערך (הרב-ממדי). כשמשתמשים ב-ByteBuffer , התוכן צריך להישאר ללא שינוי עד להשלמת ההסקה של המודל. |
---|---|
פלטים | מיפוי פלט אינדקסי פלט למערכים רב ממדיים של נתוני פלט או ל-ByteBuffer סוגים פרימיטיביים כולל int, float, long ו-byte. צריך לשמור רשומות רק כדי שאפשר יהיה להשתמש בפלטים.
|