itlab-vision / dl-benchmark

Deep Learning Inference benchmark. Supports OpenVINO™ toolkit, TensorFlow, TensorFlow Lite, ONNX Runtime, OpenCV DNN, MXNet, PyTorch, Apache TVM, ncnn, PaddlePaddle, etc.
http://hpc-education.unn.ru/dli
Apache License 2.0
27 stars 37 forks source link

[DEMO] TensorFlowLite #557

Closed Vladislav-Denisov closed 3 weeks ago

Vladislav-Denisov commented 3 weeks ago

На основе PR https://github.com/itlab-vision/dl-benchmark/pull/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

valentina-kustikova commented 3 weeks ago

@ismukhin, здесь если не будет замечаний, то тоже можете замержить следом за основным PR с демо-приложением.

ismukhin commented 3 weeks ago

@Vladislav-Denisov, подцепите, пожалуйста, актуальную версию мастера в вашу ветку, дабы конфликтов не возникло при мердже

Vladislav-Denisov commented 3 weeks ago

@Vladislav-Denisov, подцепите, пожалуйста, актуальную версию мастера в вашу ветку, дабы конфликтов не возникло при мердже

@ismukhin готово