La société Bumble, qui développe l'un des plus grands services de rencontres en ligne, a ouvert le code source du système d'apprentissage automatique Private Detector utilisé pour détecter les images obscènes dans les photos téléchargées sur le service. Le système est écrit en Python, utilise le framework Tensorflow et est distribué sous la licence Apache-2.0. Pour la classification, le réseau neuronal convolutif EfficientNet v2 est utilisé. Un modèle prêt à l'emploi pour identifier les images de personnes nues est disponible en téléchargement. La précision de détermination est supérieure à 98 %.
Il comprend également un script pour créer des modèles personnalisés qui peuvent être formés sur votre collection et utilisés pour classer un contenu arbitraire. Pour l'entraînement, il suffit d'exécuter le script avec des fichiers texte contenant des listes d'images aux propriétés positives et négatives. Une fois la formation terminée, vous pouvez transférer une image arbitraire vers le détecteur privé et, sur cette base, le taux de réussite sera calculé.
Source: opennet.ru