Обложка канала

Neural Networks | Нейронные сети

Нейронные сети в действии! Обучающие материалы и эксперименты.

Neural Networks | Нейронные сети

4 года назад
Открыть в
Опенсорс инструмент по конвертации моделей ONNX в PyTorch. Поддерживаются практически все архитектуры CV нейронок. https://github.com/ENOT-AutoDL/onnx2torch Основные сценарии использования: 1) Модификации обученной модели. Например есть обученная моделька на onnx, у неё на входе 3 канала, а надо 1. Берёшь onnx2torch, конвертишь в PyTorch, на первом слое меняешь каналы (заменяешь слой целиком), дотюниваешь. 2) Возможность доставать сетки из других фреймворков (например обученные бэкбоны).
GitHub - ENOT-AutoDL/onnx2torch: Convert onnx models to pytorch.

Convert onnx models to pytorch. Contribute to ENOT-AutoDL/onnx2torch development by creating an account on GitHub.

GitHub