nsfw_image_detection Falconsai

유형
model
태스크
image-classification
프레임워크
transformers
다운로드
4,670,569
좋아요
1,160
접근
public
파일
0

태그

  • 이미지 분류
  • 컴퓨터 비전
  • 비전 트랜스포머
  • transformers
  • AI 안전
  • 파인튜닝
  • PyTorch

요약

이 모델은 ViT(Vision Transformer, google/vit-base-patch16-224-in21k)를 기반으로 약 8만 장의 이미지로 파인튜닝하여 이미지가 NSFW(비업무용 유해 콘텐츠)인지 여부를 'normal'과 'nsfw' 두 클래스로 분류합니다. 콘텐츠 안전 및 검열 필터링에 특화되어 있으며, transformers 파이프라인으로 간편하게 사용할 수 있습니다. 성인 콘텐츠 필터, 유해 이미지 차단 등 콘텐츠 중재 시스템에 활용하기 적합합니다.

README

--- license: apache-2.0 pipeline_tag: image-classification --- # 모델 카드: NSFW 이미지 분류를 위한 파인튜닝된 비전 트랜스포머(ViT) ## 모델 설명 **파인튜닝된 비전 트랜스포머(ViT)**는 BERT와 유사한 트랜스포머 인코더 아키텍처의 변형으로, 이미지 분류 작업에 맞게 조정되었습니다. "google/vit-base-patch16-224-in21k"라는 이름의 이 특정 모델은 ImageNet-21k 데이터셋을 활용하여 대규모 이미지 컬렉션에서 지도 학습 방식으로 사전…

查看完整页面 · 查看原文