Bumble åbnede et maskinlæringssystem for at opdage obskøne billeder

Virksomheden Bumble, der udvikler en af ​​de største online datingtjenester, har åbnet kildekoden til det private detektor maskinlæringssystem, der bruges til at opdage obskøne billeder på billeder, der er uploadet til tjenesten. Systemet er skrevet i Python, bruger Tensorflow-rammeværket og distribueres under Apache-2.0-licensen. Til klassificering bruges det foldede neurale netværk EfficientNet v2. En færdiglavet model til at identificere billeder af nøgne mennesker er tilgængelig til download. Bestemmelsesnøjagtigheden er mere end 98%.

Det inkluderer også et script til at skabe brugerdefinerede modeller, der kan trænes på din samling og bruges til at klassificere vilkårligt indhold. Til træning er det nok at køre scriptet med tekstfiler, der indeholder lister over billeder med positive og negative egenskaber. Efter at træningen er afsluttet, kan du overføre et vilkårligt billede til den private detektor, og ud fra det vil hit-forholdet blive beregnet.

Kilde: opennet.ru

Tilføj en kommentar