Как количество классов влияет на скорость FPS в сети YOLO - PullRequest
2 голосов
/ 01 октября 2019

Я новичок в области компьютерного зрения и нейронных сетей. Я обучил сеть, используя Yolo-tiny, используя подмножество набора данных COCO, потому что я хотел обучить свою сеть только с одним классом (Person). Обучение прошло успешно, точность не очень хорошая (я знаю, что Yolo-tiny быстрее, но ниже, чем YOLOV3), и я получаю около 50-60FPS на видео (с размером 416x416 в файле cfg), но очень низкий FPS, если я тестирую вприложение в режиме реального времени через камеру (около 2-3FPS). При повторном обучении сети с большим количеством классов, такие показатели не улучшились. Мой вопрос в основном, как количество классов, которые мы выбираем для обучения сети, влияет на скорость / FPS обнаружения? Если я тренирую сеть против 1 класса или против 80 классов, это также приведет к улучшению скорости или нет? Если нет, то почему?

...