Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Sign up
Appearance settings

qwertyforce/anti_sus

Repository files navigation

anti_sus

Outlier detection in embeddings
download clip_visual.onnx from here
download model_wat.onnx from here

how it works

get features with image_text_features_web copy them to folder ./clean/

train_gmm.py -> trains gmm with features from ./clean
testing.ipynb -> notebook for comparing distributions of ./clean/ and ./test/ for manual adjusting of threshold
anti_sus.py -> zeromq server for filtering outlier images. Receives batch of rgb numpy images, returns indexes of good images.
It has 2 step filtering:

  • gmm score threshold
  • watermark detection (filters images with watermarks, trained on scenery_watermarks)

Docker

docker build -t qwertyforce/anti_sus_nomad:1.0.0 --network host -t qwertyforce/anti_sus_nomad:latest ./
docker run -d --network host --name anti_sus_nomad qwertyforce/anti_sus_nomad:1.0.0

About

Outlier detection in image embeddings

Resources

License

Stars

Watchers

Forks

Packages

No packages published

AltStyle によって変換されたページ (->オリジナル) /