[DEMO] Added support: MXNet, ONNX, OpenCV, PyTorch, TensorFlow, TVM #552
Conversation
881c01e to
b2895ff
Compare
32bcd37 to
8e324a1
Compare
3c5aada to
cef0665
Compare
| <Task>classification</Task> | ||
| <Name>SampleNet</Name> | ||
| <Precision>FP32</Precision> | ||
| <SourceFramework>ONNXRuntime</SourceFramework> |
There was a problem hiding this comment.
Эта модель точно не сконвертирована из Caffe в ONNX?
There was a problem hiding this comment.
Не знаю насколько это критично в рамках конфига для демо.
В офф репозитории https://github.com/openvinotoolkit/open_model_zoo/tree/master/tools/accuracy_checker/data/test_models для этих SampleNet'ов нет никакой README-шки.
There was a problem hiding this comment.
Не критично, просто если она сконвертирована, то лучше писать исходный формат.
|
@valentina-kustikova на данный момент в этом PR есть всё необходимое для демо с использованием фреймворков, перечисленных в первом сообщении. |
| <Task>classification</Task> | ||
| <Name>SampleNet</Name> | ||
| <Precision>FP32</Precision> | ||
| <SourceFramework>ONNXRuntime</SourceFramework> |
There was a problem hiding this comment.
Не критично, просто если она сконвертирована, то лучше писать исходный формат.
|
@ismukhin, посмотрите, пожалуйста. |
docker/TVM/Dockerfile
Outdated
|
|
||
|
|
||
| WORKDIR /tmp/open_model_zoo/ | ||
| RUN git remote add omz_custom_tvm https://github.com/a-sidorova/open_model_zoo.git && \ |
There was a problem hiding this comment.
Мне кажется, ветку с accuracy_checker'ом для tvm нужно форкнуть в itlab-vision, чтобы в будущем не оказалось, что его больше нет :) (думаю, сделать и держать его у нас имеет смысл до момента включения tvm в мастер open_model_zoo)
There was a problem hiding this comment.
Создала бранчу в open_model_zoo_tvm 😀 (коммиты поверх 2024.3 релиза).
https://github.com/itlab-vision/open_model_zoo_tvm/tree/24.3.0/tvm
|
@ismukhin, посмотрите, если все ок, то замержите, пожалуйста. |
На основе PR #552 ## Изменения Добавлен запуск демо-приложения для фреймворка TensorFlowLite. Оформлено отдельным PR, чтобы не задерживать мердж PR #552 ## Доп информация Изменения, которые внесены в докер-файл, связаны с тем, что при конкретной указанной версии `onnx=1.10.2` https://github.com/itlab-vision/dl-benchmark/blob/f8994ef4988abdd9df7c29063cdd8bdf2c9c9f07/docker/TensorFlowLite/Dockerfile#L20 не удается собрать образ. Возникает проблема: ``` Traceback (most recent call last): File "/usr/local/bin/cmake", line 5, in <module> from cmake import cmake ModuleNotFoundError: No module named 'cmake' ``` Попытка установить cmake через pip проблему не решает. Похожая картина остается с apt install. Причем, на первый взгляд, без проблем работает та версия, которая устанавливается согласно https://github.com/itlab-vision/dl-benchmark/blob/f8994ef4988abdd9df7c29063cdd8bdf2c9c9f07/docker/TensorFlowLite/Dockerfile#L15
Изменения:
Фреймворки:
accuracy_checker.accuracy_checker. А также непосредственно в обертку для запускаaccuracy_checkerвdl-benchmark.accuracy_checker.accuracy_checkизменен.accuracy_checker.accuracy_checkerс кастомной реализацией.