bhky/opennsfw2

해결하는 문제

이미지 및 동영상 내의 음란물 콘텐츠를 감지하기 위한 Yahoo Open-NSFW 모델의 현대적이고 사용하기 쉬운 구현체를 제공하여, 중단된 Caffe 기반의 기존 버전을 대체합니다.

작동 방식

이 프로젝트는 사전 학습된 Yahoo 모델의 Keras 3 구현을 사용합니다. TensorFlow와 JAX를 지원하는 여러 백엔드를 지원하며, 단일 이미지, 이미지 배치, 동영상 프레임을 처리하기 위한 고수준 API를를 제공합니다. 원래 모델의 가중치와 호환성을 보장하기 위해 특정 전처리 파이프라인(YAHOO 및 SIMPLE)을 포함합니다.

대상

NSFW 콘텐츠 감지를 애플리케이션에 통합해야 하는 개발자, 특히 Keras/TensorFlow/JAX 생태계를 사용하는 개발자입니다.

누구를 위한 것인가

NSFW 콘텐츠 감지를 애플리케이션에 통합해야 하는 개발자, 특히 Keras/TensorFlow/JAX 생태계를 사용하는 개발자입니다.

주요 특징

• 멀티 백엔드 지원: Keras 3 (TensorFlow/JAX) 및 tf-keras와 함께 작동합니다.

• 동영상 분석: 동영상의 모든 프레임에 대해 NSFW 확률을 예측할 수 있습니다.

• Grad-CAM 지원: 예측을 유도한 이미지 영역을 강조하는 히트맵을 생성합니다 (TensorFlow 백엔드 전용).

• 微调 (Fine-tuning): TensorFlow를 사용하여 사용자 정의 데이터셋으로 모델을 학습하거나 미세 조정하는 데 필요한 예제와 도구를 제공합니다.