Společnost Bumble, která vyvíjí jednu z největších online seznamovacích služeb, otevřela zdrojový kód pro systém strojového učení Private Detector, který slouží k detekci obscénních obrázků na fotografiích nahraných do služby. Systém je napsán v Pythonu, využívá framework Tensorflow a je distribuován pod licencí Apache-2.0. Pro klasifikaci je použita konvoluční neuronová síť EfficientNet v2. Ke stažení je k dispozici hotový model pro identifikaci obrázků nahých lidí. Přesnost stanovení je více než 98 %.
Obsahuje také skript pro vytváření vlastních modelů, které lze trénovat na vaší sbírce a používat je ke klasifikaci libovolného obsahu. Pro trénování postačí skript spustit s textovými soubory obsahujícími seznamy obrázků s pozitivními a negativními vlastnostmi. Po dokončení tréninku můžete do Soukromého detektoru přenést libovolný obrázek a na základě něj bude vypočten poměr zásahů.
Zdroj: opennet.ru