Skip to content

Commit

Permalink
add violence detection prompts
Browse files Browse the repository at this point in the history
Thanks to https://github.com/ml-research/OffImgDetectionCLIP
usage: first embedding is safe, second is unsafe
compute similarity between your embeddings and these two, if the closest is the safe embedding then it's safe, otherwise it's unsafe
these have been trained on https://osf.io/2rqad/ and are good at detecting violent content
  • Loading branch information
rom1504 authored Apr 10, 2022
1 parent 057de24 commit c2f21fe
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 0 deletions.
Binary file added violence_detection_vit_b_32.npy
Binary file not shown.
Binary file added violence_detection_vit_l_14.npy
Binary file not shown.

0 comments on commit c2f21fe

Please sign in to comment.