Оптимизация YoloV8 для смартфона

Question

Я использую модель YoloV8 с tensorflow.js. Я пытаюсь выполнить сегментацию изображений в реальном времени с помощью камеры.

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

Приложение сегментации объектов с использованием YOLOv8 и Tensorflow.js

Вместо того, чтобы обнаруживать 80 типов объектов, возможно, я мог бы оптимизировать его только для 2 или 3 объектов. Есть ли такая возможность в tensorflow.js или мне нужно тренироваться с новым набором данных?

Спасибо за помощь

Answer 1

Я рекомендую вам экспортировать вашу модель в дружественную для Android среду, такую ​​как TFLite. Это даст вам ~10 кадров в секунду на приличном процессоре телефона.sмодель. Вы можете еще больше увеличить FPS, экспортировав в TFLite INT8. Это даст вам прирост только в том случае, если ЦП/ГП поддерживает операции INT8.