itlab-vision / dl-benchmark

Deep Learning Inference benchmark. Supports OpenVINO™ toolkit, Caffe, TensorFlow, TensorFlow Lite, ONNX Runtime, OpenCV DNN, MXNet, PyTorch, Apache TVM, ncnn, etc.

Home Page:http://hpc-education.unn.ru/dli

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Объединение функций обработки выхода модели

valentina-kustikova opened this issue · comments

@roma-roma97, @TheG1uy (@FenixFly, @KchnKchn), в скриптах inference_sync_mode.py и inference_async_mode.py есть функции обработки выхода нейросетевых моделей в зависимости от задачи. Можно ли эти функции вынести в отдельный модуль (очень похоже, что код идентичен и не зависит от режима запуска)? Если можно, то надо это сделать, поскольку сейчас у нас похоже присутствует дублирование кода.

Да. Код идентичен. Сегодня исправлю.

Вынес все инварианты в модули. Жду ревью кода.
Задача выполнена!