Как нарисовать ограничивающие рамки для прогнозируемых классов для пользовательского обученного классификатора изображений с использованием fastai

Question

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

Answer 1

Так что я бы сделал это в два этапа. Сначала узнайте, как получить ограничивающие рамки в фастай. Пример здесь .

Если у вас есть, вы можете использовать opencv для загрузки видео по кадрам и пометить его. Пример OpenCV .

Когда у вас есть оба, это довольно легко сделать в реальном времени. Просто пометьте кадры, загруженные с вашей веб-камеры. Этот процесс довольно дорогой, поэтому вы, вероятно, захотите снизить разрешение изображения перед выводом. Я сделал это в тензорном потоке, и он работал довольно хорошо. Этот репозиторий был чрезвычайно полезен