Bumble åpner maskinlæringssystem for å oppdage usømmelige bilder

Bumble, som utvikler en av de største online datingtjenestene, har åpnet kildekoden til Private Detectors maskinlæringssystem, som brukes til å identifisere usømmelige bilder i bilder lastet opp til tjenesten. Systemet er skrevet i Python, bruker Tensorflow-rammeverket og distribueres under Apache-2.0-lisensen. EfficientNet v2 konvolusjonelle nevrale nettverk brukes til klassifisering. En ferdig modell for å identifisere bilder av nakne mennesker er tilgjengelig for nedlasting. Nøyaktigheten av bestemmelsen er mer enn 98%.

Det inkluderer også et skript for å lage dine egne modeller, som du kan trene på samlingen din og bruke til å klassifisere vilkårlig innhold. For trening er det nok å kjøre et skript med tekstfiler som inneholder lister over bilder med positive og negative egenskaper. Etter at treningen er fullført, kan du sende et vilkårlig bilde til Private Detector og treffraten vil bli beregnet basert på det.

Kilde: opennet.ru

Legg til en kommentar