คลาส 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 เพียงแต่ต้องเก็บรายการไว้สำหรับเอาต์พุตที่จะนำมาใช้
|