Skip to content

Converter between the models of the frameworks [ONNX Torch TFLite]

Notifications You must be signed in to change notification settings

vasabi-root/Converter

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Machine Learning Model Converter

Устанавливаем CMake. Клоним репо. Устанавливаем requirements.txt:

git clone https://github.com/vasabi-root/Converter.git
cd Converter
python -m venv .venv
.venv/Scripts/activate
pip install -r requirements.txt

В этом файле уже указаны мои форки с фиксами некоторых либ.

Далее рекомендую ознакомиться с доками.

Если кому-то потребуется коммитить, то вот инфа о том, какие либы я менял и как их собрать:

  • onnx2torch
    Качаем и устанавливаем:
    git clone https://github.com/vasabi-root/onnx2torch.git
    cd onnx2torch
    python -m pip install --upgrade build
    python -m build
    cd dist
    pip install onnx2torch-.....whl
  • onnx2tf. Не коммитил, ибо решение некорректное (подробнее в доках). Но модифицированные файлы лежат в папке changes/onnx2tf

About

Converter between the models of the frameworks [ONNX Torch TFLite]

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages