ML Bench

Machine Learning Benchmark Tool (ML Bench) (AI Benchmark Tool)

รุ่นที่รองรับ:

- MobileNet v1.0

- MobileNet v2.0

- เริ่มต้น v3.0

- Resnet v2 50

- SSD Mobilenet v1 (การตรวจจับวัตถุ)

รันไทม์ที่รองรับ:

- Tensorflow Lite

- Tensorflow มือถือ

- Android NN

- SNPE (สำหรับ Qualcomm)

- PyTorth

- TVM

วิธีใส่โมเดลของคุณ:

1. เตรียมไฟล์โมเดลของคุณ (ประเภทที่รองรับ ได้แก่ tflite, pytorch, tvm, dlc)

2. บนเครื่องของคุณสร้างไดเร็กทอรี [Model Name]

3. คัดลอกไฟล์โมเดลของคุณไปยังไดเร็กทอรี [Model Name]

4. สร้างไฟล์ชื่อ meta-data.json ในไดเร็กทอรี [Model Name]

ตัวอย่างของ meta-data.json:

{

"xres": 299,

"ปี": 299,

"ความลึก": 3,

"input_type": "ลอย",

"output_type": "ลอย",

"input_name": "อินพุต: 0",

"output_name": "InceptionV3 / Predictions / Reshape_1: 0",

"image_mean": 0,

"image_std": 0,

"label_offset": 1

}

* คุณยังสามารถตั้งค่าป้ายกำกับความจริงพื้นและไฟล์ภาพของคุณเองได้

{

...

"ป้ายกำกับ": "my_labels.txt",

"ground_truth": "my_ground_truth.txt",

"image_path": "ของฉัน / path / to / images",

...

}

5. พุชไดเร็กทอรี [ชื่อรุ่น] ไปยังอุปกรณ์ของคุณโดยใช้คำสั่งด้านล่าง

adb push ./ [ชื่อรุ่น] / sdcard / ดาวน์โหลด / MlBenchModels /