ML Bench

Công cụ điểm chuẩn máy học (ML Bench) (Công cụ điểm chuẩn AI)

Các mô hình được hỗ trợ:

- MobileNet v1

- MobileNet v2

- Khởi đầu v3

- Resnet v2 50

- SSD Mobilenet v1 (Phát hiện đối tượng)

Thời gian chạy được hỗ trợ:

- Tensorflow Lite

- Tensorflow Mobile

- Android NN

- SNPE (dành cho Qualcomm)

- PyTorth

- TVM

Cách tải mô hình của bạn bên cạnh:

1. Chuẩn bị tệp mô hình của bạn (các loại được hỗ trợ là tflite, pytorch, tvm, dlc)

2. Trên máy cục bộ của bạn, tạo thư mục [Tên kiểu máy]

3. Sao chép tệp mô hình của bạn vào thư mục [Tên mô hình]

4. Tạo một tệp có tên là meta-data.json trong thư mục [Tên kiểu]

ví dụ về meta-data.json:

{

"xres": 299,

"yres": 299,

"độ sâu": 3,

"input_type": "float",

"output_type": "float",

"input_name": "input: 0",

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

"image_mean": 0,

"image_std": 0,

"label_offset": 1

}

* bạn cũng có thể đặt nhãn, sự thật nền và các tệp hình ảnh của riêng bạn

{

...

"label": "my_labels.txt",

"ground_truth": "my_ground_truth.txt",

"image_path": "my / path / to / images",

...

}

5. đẩy thư mục [Tên kiểu] vào thiết bị của bạn bằng lệnh dưới đây

adb push ./[Model Name] / sdcard / Download / MlBenchModels /