คลาส Wrapper สำหรับโมเดล TFLite และล่ามแบบ TFLite
หมายเหตุ: Model จะเก็บโมเดล TFLite ได้เพียงครั้งละ 1 รายการ และจะเก็บอินสแตนซ์อินเทอร์พรีเตอร์ TFLite เพื่อเรียกใช้เสมอ
ชั้นเรียนที่ซ้อนกัน
| คลาส | Model.Builder |
ชั้นเรียนนี้เลิกใช้งานแล้ว โปรดใช้ Model.createModel(Context, String, Options)
|
|
| enum | Model.Device | ประเภทอุปกรณ์รันไทม์ที่ใช้สําหรับการแยกประเภท | |
| คลาส | Model.Options | ตัวเลือกสำหรับการเรียกใช้โมเดล | |
วิธีการสาธารณะ
| void |
close()
|
| โมเดลแบบคงที่ |
createModel(บริบทบริบท, ตัวเลือก String modelPath, Model.Options)
โหลดโมเดลจากเนื้อหาและเริ่มต้นอินเทอร์พรีเตอร์ TFLite ด้วยตัวเลือกที่มีให้
|
| โมเดลแบบคงที่ | |
| MappedByteBuffer |
getData()
แสดงผลข้อมูลโมเดลการแมปหน่วยความจำ
|
| Tensor |
getInputTensor(int InputIndex)
รับ Tensor ที่เชื่อมโยงกับดัชนีอินพุตที่ระบุ
|
| Tensor |
getOutputTensor(intเอาต์พุตIndex)
รับ Tensor ที่เชื่อมโยงกับดัชนีเอาต์พุตที่ระบุ
|
| int[] |
getOutputTensorShape(int exportIndex)
แสดงผลรูปร่างเอาต์พุต
|
| String |
getPath()
แสดงผลเส้นทางของไฟล์โมเดลที่จัดเก็บไว้ในชิ้นงาน
|
| void |
วิธีการที่รับช่วงมา
วิธีการสาธารณะ
สาธารณะ โมฆะ ปิด ()
สาธารณะ คงที่ โมเดล createModel (บริบทบริบท, String modelPath, Model.Options ตัวเลือก)
โหลดโมเดลจากเนื้อหาและเริ่มต้นอินเทอร์พรีเตอร์ TFLite ด้วยตัวเลือกที่มีให้
พารามิเตอร์
| บริบท | บริบทของแอป |
|---|---|
| modelPath | เส้นทางของไฟล์โมเดล |
| ตัวเลือก | ตัวเลือกในการเรียกใช้โมเดล |
การขว้าง
| IOException | หากมีข้อยกเว้นเกิดขึ้นเมื่อเปิดไฟล์โมเดล |
|---|
ดูเพิ่มเติม
สาธารณะ คงที่ โมเดล createModel (บริบทบริบท, String modelPath)
โหลดโมเดลจากชิ้นงานและเริ่มต้นอินเทอร์พรีเตอร์ TFLite
ตัวเลือกเริ่มต้นคือ (1) อุปกรณ์ CPU (2) 1 เทรด
พารามิเตอร์
| บริบท | บริบทของแอป |
|---|---|
| modelPath | เส้นทางของไฟล์โมเดล |
การขว้าง
| IOException | หากมีข้อยกเว้นเกิดขึ้นเมื่อเปิดไฟล์โมเดล |
|---|
สาธารณะ Tensor getInputTensor (int InputIndex)
รับ Tensor ที่เชื่อมโยงกับดัชนีอินพุตที่ระบุ
พารามิเตอร์
| inputIndex |
|---|
การขว้าง
| IllegalStateException | หากล่ามปิดอยู่ |
|---|
สาธารณะ Tensor getOutputTensor (int exportIndex)
รับ Tensor ที่เชื่อมโยงกับดัชนีเอาต์พุตที่ระบุ
พารามิเตอร์
| outputIndex |
|---|
การขว้าง
| IllegalStateException | หากล่ามปิดอยู่ |
|---|
public int[] getOutputTensorShape (int exportIndex)
แสดงผลรูปร่างเอาต์พุต มีประโยชน์หากกำหนดรูปร่างเอาต์พุตเมื่อมีการสร้างกราฟเท่านั้น
พารามิเตอร์
| outputIndex |
|---|
การขว้าง
| IllegalStateException | หากล่ามปิดอยู่ |
|---|
สาธารณะ void run (อินพุต ออบเจ็กต์[], เอาต์พุต แมป<จำนวนเต็ม, Object>)
เรียกใช้การอนุมานโมเดลกับอินพุตหลายรายการและแสดงผลเอาต์พุตหลายรายการ
พารามิเตอร์
| อินพุต | อาร์เรย์ของข้อมูลอินพุต อินพุตควรอยู่ในลำดับเดียวกับอินพุตของโมเดล อินพุตแต่ละรายการอาจเป็นอาร์เรย์หรืออาร์เรย์หลายมิติ หรือ ByteBuffer ของประเภทดั้งเดิม ซึ่งรวมถึง int, Float, Long และ Byte ByteBuffer เป็นวิธีที่แนะนำในการส่งข้อมูลอินพุตขนาดใหญ่ ในขณะที่ประเภทสตริงจำเป็นต้องใช้เส้นทางอินพุตอาร์เรย์ (หลายมิติข้อมูล) เมื่อใช้ ByteBuffer เนื้อหาของจะไม่มีการเปลี่ยนแปลงจนกว่าจะทำการอนุมานโมเดลเสร็จ |
|---|---|
| เอาต์พุต | ดัชนีเอาต์พุตการแมปแผนที่ไปยังอาร์เรย์หลายมิติของข้อมูลเอาต์พุต หรือ ByteBuffer ของประเภทดั้งเดิม ซึ่งรวมถึง int, Float, Long และ Byte เพียงแต่ต้องเก็บรายการไว้สำหรับเอาต์พุตที่จะนำมาใช้
|