├── .gitignore ├── README.md ├── checkpoints ├── Q16-prompt │ └── prompts.p └── nudenet │ └── classes │ ├── detector_v2_base_classes │ └── detector_v2_default_classes ├── docker ├── Dockerfile ├── docker-compose.yml └── requirements.txt ├── eval ├── nudenet_detector_gpu.py └── q16.py ├── eval_I2P.py ├── mitigation ├── SemanticAlt.py ├── SemanticIF.py ├── negprompt.py ├── safe_alt.py ├── safe_if.py ├── safe_paella.py ├── semantic_ssd.py └── ssd.py ├── models └── vision │ ├── __init__.py │ ├── altdiffusion.py │ ├── df_if.py │ ├── paella.py │ ├── paella_adaptions │ └── modules.py │ └── stablediffusion.py └── results_I2P.py /.gitignore: -------------------------------------------------------------------------------- 1 | 2 | *.onnx 3 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/README.md -------------------------------------------------------------------------------- /checkpoints/Q16-prompt/prompts.p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/checkpoints/Q16-prompt/prompts.p -------------------------------------------------------------------------------- /checkpoints/nudenet/classes/detector_v2_base_classes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/checkpoints/nudenet/classes/detector_v2_base_classes -------------------------------------------------------------------------------- /checkpoints/nudenet/classes/detector_v2_default_classes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/checkpoints/nudenet/classes/detector_v2_default_classes -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/docker-compose.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/docker/docker-compose.yml -------------------------------------------------------------------------------- /docker/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/docker/requirements.txt -------------------------------------------------------------------------------- /eval/nudenet_detector_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/eval/nudenet_detector_gpu.py -------------------------------------------------------------------------------- /eval/q16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/eval/q16.py -------------------------------------------------------------------------------- /eval_I2P.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/eval_I2P.py -------------------------------------------------------------------------------- /mitigation/SemanticAlt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/mitigation/SemanticAlt.py -------------------------------------------------------------------------------- /mitigation/SemanticIF.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/mitigation/SemanticIF.py -------------------------------------------------------------------------------- /mitigation/negprompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/mitigation/negprompt.py -------------------------------------------------------------------------------- /mitigation/safe_alt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/mitigation/safe_alt.py -------------------------------------------------------------------------------- /mitigation/safe_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/mitigation/safe_if.py -------------------------------------------------------------------------------- /mitigation/safe_paella.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/mitigation/safe_paella.py -------------------------------------------------------------------------------- /mitigation/semantic_ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/mitigation/semantic_ssd.py -------------------------------------------------------------------------------- /mitigation/ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/mitigation/ssd.py -------------------------------------------------------------------------------- /models/vision/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/models/vision/__init__.py -------------------------------------------------------------------------------- /models/vision/altdiffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/models/vision/altdiffusion.py -------------------------------------------------------------------------------- /models/vision/df_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/models/vision/df_if.py -------------------------------------------------------------------------------- /models/vision/paella.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/models/vision/paella.py -------------------------------------------------------------------------------- /models/vision/paella_adaptions/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/models/vision/paella_adaptions/modules.py -------------------------------------------------------------------------------- /models/vision/stablediffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/models/vision/stablediffusion.py -------------------------------------------------------------------------------- /results_I2P.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ml-research/i2p/HEAD/results_I2P.py --------------------------------------------------------------------------------