Skip to content

[DEMO] Added support: MXNet, ONNX, OpenCV, PyTorch, TensorFlow, TVM

Valentina Kustikova requested to merge github/fork/Vladislav-Denisov/demo into master

Created by: Vladislav-Denisov

Изменения:

  • Внесены общие изменения в логику запуска демо (для поддержки запуска различных фреймворков и переиспользования кода).

Фреймворки:

  • MXNet - добавлен в демо. Внесены правки в docker-файл для проверки работоспособности accuracy_checker.
  • ONNX - добавлен в демо. Внесены правки в docker-файл для проверки работоспособности accuracy_checker. А также непосредственно в обертку для запуска accuracy_checker в dl-benchmark.
  • OpenCV - добавлен в демо. Внесены правки в docker-файл для проверки работоспособности accuracy_checker.
  • TensorFlow - добавлен в демо на примере запуска для модели фреймворка TF2. Лаунчер для accuracy_check изменен.
  • PyTorch - добавлен в демо. Внесены правки в docker-файл для проверки работоспособности accuracy_checker.
  • TVM - добавлен в демо. Внесены правки в docker-файл для поддержки accuracy_checker с кастомной реализацией.

Merge request reports