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 /
Adnroid 12 support