Bumble отвори система за машинно обучение за откриване на неприлични изображения

Компанията Bumble, която разработва една от най-големите онлайн услуги за запознанства, отвори изходния код на системата за машинно обучение Private Detector, използвана за откриване на неприлични изображения в снимки, качени в услугата. Системата е написана на Python, използва рамката Tensorflow и се разпространява под лиценз Apache-2.0. За класификация се използва конволюционната невронна мрежа EfficientNet v2. Готов модел за разпознаване на изображения на голи хора е достъпен за изтегляне. Точността на определяне е повече от 98%.

Той също така включва скрипт за създаване на потребителски модели, които могат да бъдат обучени във вашата колекция и използвани за класифициране на произволно съдържание. За обучение е достатъчно да стартирате скрипта с текстови файлове, съдържащи списъци с изображения с положителни и отрицателни свойства. След приключване на обучението можете да прехвърлите произволно изображение към частния детектор и въз основа на него да се изчисли коефициентът на попадение.

Източник: opennet.ru