Разработчик Пит Уорден впервые за три года выпустил обновление порта TensorFlow Lite для плат Raspberry Pi Pico на базе микроконтроллеров RP2040. В релиз включили изменения, которые накопились за это время, и добавили оптимизированную работу на двух ядрах.
Пит Уорден начал разработку порта TensorFlow Lite в 2020 году, когда он работал в Google над мобильной версией TensorFlow. После этого развитие проекта остановилось. Спустя три года Уорден перешёл в компанию Useful Sensors и снова взялся за TensorFlow Lite для Raspberry Pi Pico.
В обновление включили ускоренную реализацию CMSIS-NN для Conv2D, разделённую на два ядра RP2040. При этом разработчик отмечает, что из-за этого начали появляться ошибки переполнения стека в некоторых тестах. Поэтому оптимизацию можно отключить. В блоге Raspberry Pi отметили, что обновление ускорило время прохождения теста по обнаружению человека с 824 мс до 588 мс.
Уорден рассказал, что окончательно покинул Google, поэтому теперь будет активно развивать порт TensorFlow Lite для Raspberry Pi Pico. Вместе с этим, он встроил скрипты, которые будут автоматически включать обновления из основного репозитория TensorFlow в репозиторий порта. Благодаря этому разработчикам всегда будут доступны свежие версии. Более подробно о работе над проектом Уорден рассказывает в своём блоге.
Источник: habr.com