Skip to content

mlpack/onnx-mlpack

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ONNX-mlpack Translator

ONNX-mlpack Translator

Unlock the Power of Other Frameworks in mlpack

This repository contains a converter for certain machine learning models from onnx to mlpack format. Currently this repository is still under construction and might undergo some major refactoring, please use with cautios.

Dependencies of this repository

  • Protocol Buffer
  • ONNX

ONNX internally download and build protobuf for ONNX build. So you just need to build ONNX. For building onnx follow their [ReadMe](ONNX will internally download and build protobuf for ONNX build.).

The repository is in developing phase and its been tested on the following models.

Models Model Generation Weight Transfer
Iris classification ✔️ ✔️
Yolo ✔️ ✔️