Мобильное приложение для тестирования моделей машинного обучения (TFLite, PyTorch)

Работая над системой, которая позволяет каждому легко обучить нейронную сеть компьютерного зрения, наша команда ML создала приложение для проверки того, какая модель работает лучше всего на данном мобильном устройстве.

Прежде чем разрабатывать собственное приложение на основе ML для распознавания и классификации изображений, пользователям необходимо выбрать модель машинного обучения для обучения. На этот выбор в значительной степени влияет мобильное устройство, на котором будет работать приложение, поскольку более продвинутые модели работают быстрее на новых устройствах и медленнее на старых. Чтобы помочь пользователям найти модель, которая будет работать быстрее всего на их мобильном устройстве, мы разработали простое, но эффективное приложение для тестирования моделей.

Мы адаптировали четыре модели распознавания и классификации изображений TensorFlow Lite и три модели PyTorch для использования на устройстве Android. Пользователи могут выбрать один из двух размеров изображений, чтобы лучше отразить реальные условия работы, то есть размер изображений в реальном наборе данных. После этого приложение тестирует каждую модель и оценивает их в соответствии с их производительностью.

Приложение позволяет пользователям в считанные минуты сделать правильный выбор между различными моделями TensorFlow Lite и PyTorch, обеспечивая максимальную эффективность их проекта.