Выложенная в открытый доступ библиотека CerberusDet позволяет обучать модели компьютерного зрения для нескольких задач обнаружения объектов.

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

Библиотека содержит как код обучения и инференса, так и созданные с помощью фреймворка модели CerberusDet на основе архитектуры YOLO.

Решение CerberusDet было представлено разработчиками из SberDevices в июле 2024 года.

Препринт на arXiv

👤 Максим Купрашевич

👤 Ирина Толстых