Skip to content

Repository files navigation

Filter Images by CLIP prompts

Script filter_images.py permite determinar qué imágenes coinciden con una descripción positiva y censurar las que coincidan con prompts negativos (temas sensibles).

Uso básico:

  1. Activar tu entorno virtual y asegurarte de tener torch y transformers instalados.
  2. Ejecutar:
python filter_images.py --desc "A small dog with a ball" --images "ejemplo*.jpg" --out res.json

Parámetros importantes:

  • --desc: descripción positiva obligatoria.
  • --images: glob pattern para las imágenes (por defecto ejemplo*.jpg).
  • --negatives: lista opcional de prompts negativos.
  • --pos_thresh / --neg_thresh: umbrales para aceptar o censurar.

Salida: JSON con resultados por imagen.

About

compare a description with multiple images

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages