โมเดล ชั้นเรียนสาธารณะ
คลาส wrapper สำหรับโมเดล TFLite และล่าม TFLite
หมายเหตุ: Model สามารถเก็บโมเดล TFLite ได้ครั้งละ 1 รายการเท่านั้น และจะเก็บอินสแตนซ์ล่าม TFLite เพื่อเรียกใช้เสมอ
คลาสที่ซ้อนกัน
| ระดับ | โมเดล.บิวเดอร์ | ชั้นเรียนนี้เลิกใช้แล้ว กรุณาใช้ Model.createModel(Context, String, Options) | |
| แจกแจง | รุ่น.อุปกรณ์ | ประเภทอุปกรณ์รันไทม์ที่ใช้สำหรับดำเนินการจัดหมวดหมู่ | |
| ระดับ | รุ่น ตัวเลือก | ตัวเลือกสำหรับการรันโมเดล | |
วิธีการสาธารณะ
| เป็นโมฆะ | ปิด () |
| โมเดล คงที่ | createModel (บริบทบริบท, สตริง modelPath, ตัวเลือก Model.Options ) โหลดโมเดลจากเนื้อหาและเริ่มต้นล่าม TFLite ด้วยตัวเลือกที่กำหนด |
| โมเดล คงที่ | |
| แมป ByteBuffer | รับข้อมูล () ส่งคืนข้อมูลแบบจำลองที่แมปหน่วยความจำ |
| เทนเซอร์ | getInputTensor (int inputIndex) รับเทนเซอร์ที่เกี่ยวข้องกับดัชนีอินพุตที่ให้มา |
| เทนเซอร์ | getOutputTensor (int เอาท์พุทดัชนี) รับค่าเทนเซอร์ที่เกี่ยวข้องกับดัชนีเอาต์พุตที่ระบุ |
| อินท์[] | getOutputTensorShape (int เอาท์พุตดัชนี) ส่งกลับรูปร่างผลลัพธ์ |
| สตริง | รับเส้นทาง () ส่งกลับเส้นทางของไฟล์โมเดลที่จัดเก็บไว้ในสินทรัพย์ |
| เป็นโมฆะ |
วิธีการสืบทอด
วิธีการสาธารณะ
โมฆะสาธารณะ ปิด ()
โมเดล คงที่สาธารณะ createModel (บริบทบริบท, สตริง modelPath, ตัวเลือก Model.Options )
โหลดโมเดลจากเนื้อหาและเริ่มต้นล่าม TFLite ด้วยตัวเลือกที่กำหนด
พารามิเตอร์
| บริบท | บริบทของแอป |
|---|---|
| โมเดลพาธ | เส้นทางของไฟล์โมเดล |
| ตัวเลือก | ตัวเลือกสำหรับการรันโมเดล |
ขว้าง
| IOข้อยกเว้น | หากมีข้อยกเว้นใดๆ เกิดขึ้นเมื่อเปิดไฟล์โมเดล |
|---|
ดูสิ่งนี้ด้วย
โมเดล คงที่สาธารณะ createModel (บริบทบริบท, สตริง modelPath)
โหลดโมเดลจากเนื้อหาและเริ่มต้นล่าม TFLite
ตัวเลือกเริ่มต้นคือ: (1) อุปกรณ์ CPU; (2) หนึ่งเธรด
พารามิเตอร์
| บริบท | บริบทของแอป |
|---|---|
| โมเดลพาธ | เส้นทางของไฟล์โมเดล |
ขว้าง
| IOข้อยกเว้น | หากมีข้อยกเว้นใดๆ เกิดขึ้นเมื่อเปิดไฟล์โมเดล |
|---|
เทนเซอร์ สาธารณะ getInputTensor (int inputIndex)
รับเทนเซอร์ที่เกี่ยวข้องกับดัชนีอินพุตที่ให้มา
พารามิเตอร์
| ดัชนีอินพุต |
|---|
ขว้าง
| IllegalStateException | ถ้าล่ามปิดอยู่ |
|---|
สาธารณะ Tensor getOutputTensor (int outputIndex)
รับค่าเทนเซอร์ที่เกี่ยวข้องกับดัชนีเอาต์พุตที่ระบุ
พารามิเตอร์
| ดัชนีเอาท์พุท |
|---|
ขว้าง
| IllegalStateException | ถ้าล่ามปิดอยู่ |
|---|
int สาธารณะ [] getOutputTensorShape (int outputIndex)
ส่งกลับรูปร่างผลลัพธ์ มีประโยชน์หากกำหนดรูปร่างเอาต์พุตเมื่อมีการสร้างกราฟเท่านั้น
พารามิเตอร์
| ดัชนีเอาท์พุท |
|---|
ขว้าง
| IllegalStateException | ถ้าล่ามปิดอยู่ |
|---|
การรัน โมฆะสาธารณะ (อินพุต Object[] , แผนที่ < Integer , Object > เอาต์พุต)
รันการอนุมานโมเดลบนอินพุตหลายตัว และส่งกลับเอาต์พุตหลายตัว
พารามิเตอร์
| อินพุต | อาร์เรย์ของข้อมูลอินพุต อินพุตควรอยู่ในลำดับเดียวกันกับอินพุตของโมเดล อินพุตแต่ละรายการสามารถเป็นอาร์เรย์หรืออาร์เรย์หลายมิติ หรือ ByteBuffer ประเภทดั้งเดิมซึ่งรวมถึง int, float, long และ byte ByteBuffer เป็นวิธีที่แนะนำในการส่งข้อมูลอินพุตขนาดใหญ่ ในขณะที่ประเภทสตริงจำเป็นต้องใช้เส้นทางอินพุตอาร์เรย์ (หลายมิติ) เมื่อใช้ ByteBuffer เนื้อหาไม่ควรเปลี่ยนแปลงจนกว่าการอนุมานโมเดลจะเสร็จสิ้น |
|---|---|
| เอาท์พุท | ดัชนีเอาต์พุตการแมปแผนที่ไปยังอาร์เรย์หลายมิติของข้อมูลเอาต์พุตหรือ ByteBuffer ของประเภทดั้งเดิมรวมถึง int, float, long และ byte เพียงแต่ต้องเก็บรายการไว้สำหรับเอาต์พุตที่จะใช้ |