search

ctuning / ck-mlperf

This repository is outdated! Join the open MLPerf workgroup to participate in the development of the next generation of automation workflows for MLPerf benchmarks:
https://bit.ly/mlperf-edu-wg
BSD 3-Clause "New" or "Revised" License
32 stars 23 forks source link

Implement MLPerf image classification program for ONNX #3

Closed psyhtest closed 5 years ago

psyhtest commented 5 years ago

We should implement cl-mlperf:program:image-classification-onnx similar to ck-tensorflow:program:image-classification-tflite and friends to support MLPerf image classification models in the ONNX format (e.g. ResNet50). Perhaps main.py already does all we need, just needs to have its inputs and outputs unified with program:image-classification-tflite.

ens-lg4 commented 5 years ago

Done, see program:image-classification-onnx-py . It supports both NHWC and NCHW models (tested).