Skip to content

Latest commit

 

History

History
26 lines (23 loc) · 1.22 KB

readme.md

File metadata and controls

26 lines (23 loc) · 1.22 KB

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