Bumble відкрив систему машинного навчання для виявлення непристойних зображень

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

До складу входить скрипт для створення власних моделей, які можна навчити на своїй колекції та використовувати для класифікації довільного вмісту. Для навчання достатньо запустити скрипт із текстовими файлами, що містять списки зображень із позитивними та негативними властивостями. Після закінчення навчання можна передати в Private Detector довільне зображення і на його основі буде обчислено коефіцієнт влучення.

Джерело: opennet.ru

Додати коментар або відгук