├── .idea ├── .gitignore ├── AnomalyDiffusion.iml ├── deployment.xml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── LICENSE ├── README.md ├── cal_ic_lpips.py ├── configs ├── autoencoder │ ├── autoencoder_kl_16x16x16.yaml │ ├── autoencoder_kl_32x32x4.yaml │ ├── autoencoder_kl_64x64x3.yaml │ └── autoencoder_kl_8x8x64.yaml ├── latent-diffusion │ ├── celebahq-ldm-vq-4.yaml │ ├── cin-ldm-vq-f8.yaml │ ├── cin256-v2.yaml │ ├── ffhq-ldm-vq-4.yaml │ ├── lsun_bedrooms-ldm-vq-4.yaml │ ├── lsun_churches-ldm-kl-8.yaml │ ├── txt2img-1p4B-eval.yaml │ ├── txt2img-1p4B-eval_with_tokens.yaml │ ├── txt2img-1p4B-finetune-encoder+embedding.yaml │ ├── txt2img-1p4B-finetune.yaml │ └── txt2img-1p4B-finetune_style.yaml └── stable-diffusion │ ├── v1-finetune.yaml │ ├── v1-finetune_unfrozen.yaml │ └── v1-inference.yaml ├── environment.yaml ├── evaluation └── clip_eval.py ├── generate_mask.py ├── generate_with_mask.py ├── ldm ├── __init__.py ├── data │ ├── __init__.py │ ├── base.py │ ├── imagenet.py │ ├── lsun.py │ ├── mask_generate.py │ ├── personalized.py │ ├── personalized_style.py │ └── tmp.jpg ├── lr_scheduler.py ├── models │ ├── autoencoder.py │ ├── diffusion │ │ ├── __init__.py │ │ ├── classifier.py │ │ ├── ddim.py │ │ ├── ddpm.py │ │ └── plms.py │ ├── encoders.py │ ├── psp_encoder.zip │ ├── psp_encoder │ │ ├── encoders │ │ │ ├── __init__.py │ │ │ ├── helpers.py │ │ │ ├── model_irse.py │ │ │ └── psp_encoders.py │ │ └── stylegan2 │ │ │ ├── __init__.py │ │ │ ├── model.py │ │ │ └── op │ │ │ ├── __init__.py │ │ │ ├── fused_act.py │ │ │ ├── fused_bias_act.cpp │ │ │ ├── fused_bias_act_kernel.cu │ │ │ ├── upfirdn2d.cpp │ │ │ ├── upfirdn2d.py │ │ │ └── upfirdn2d_kernel.cu │ └── vit.py ├── modules │ ├── attention.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── model.py │ │ ├── openaimodel.py │ │ └── util.py │ ├── distributions │ │ ├── __init__.py │ │ └── distributions.py │ ├── ema.py │ ├── embedding_manager.py │ ├── embedding_manager2.py │ ├── encoders │ │ ├── __init__.py │ │ ├── modules.py │ │ └── modules_bak.py │ ├── image_degradation │ │ ├── __init__.py │ │ ├── bsrgan.py │ │ ├── bsrgan_light.py │ │ ├── utils │ │ │ └── test.png │ │ └── utils_image.py │ ├── losses │ │ ├── __init__.py │ │ ├── contperceptual.py │ │ └── vqperceptual.py │ └── x_transformer.py ├── my_model │ ├── models.py │ └── op │ │ ├── __init__.py │ │ ├── conv2d_gradfix.py │ │ ├── fused_act.py │ │ ├── fused_bias_act.cpp │ │ ├── fused_bias_act_kernel.cu │ │ ├── upfirdn2d.cpp │ │ ├── upfirdn2d.py │ │ └── upfirdn2d_kernel.cu └── util.py ├── main.py ├── models ├── first_stage_models │ ├── kl-f16 │ │ └── config.yaml │ ├── kl-f32 │ │ └── config.yaml │ ├── kl-f4 │ │ └── config.yaml │ ├── kl-f8 │ │ └── config.yaml │ ├── vq-f16 │ │ └── config.yaml │ ├── vq-f4-noattn │ │ └── config.yaml │ ├── vq-f4 │ │ └── config.yaml │ ├── vq-f8-n256 │ │ └── config.yaml │ └── vq-f8 │ │ └── config.yaml └── ldm │ ├── bsr_sr │ └── config.yaml │ ├── celeba256 │ └── config.yaml │ ├── cin256 │ └── config.yaml │ ├── ffhq256 │ └── config.yaml │ ├── inpainting_big │ └── config.yaml │ ├── layout2img-openimages256 │ └── config.yaml │ ├── lsun_beds256 │ └── config.yaml │ ├── lsun_churches256 │ └── config.yaml │ ├── semantic_synthesis256 │ └── config.yaml │ ├── semantic_synthesis512 │ └── config.yaml │ └── text2img256 │ └── config.yaml ├── name-anomaly.txt ├── run-mvtec.py ├── scripts ├── download_first_stages.sh ├── download_models.sh ├── evaluate_model.py ├── inpaint.py ├── latent_imagenet_diffusion.ipynb ├── sample_diffusion.py ├── stable_txt2img.py └── txt2img.py ├── taming └── modules │ └── autoencoder │ └── lpips │ └── vgg.pth ├── test-classification.py ├── test-localization.py ├── train-classification.py ├── train-localization.py ├── train_mask.py ├── unet_utils ├── au_pro_util.py ├── data_loader.py ├── loss.py ├── model_unet.py └── tensorboard_visualizer.py └── utils.py /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /.idea/AnomalyDiffusion.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/.idea/AnomalyDiffusion.iml -------------------------------------------------------------------------------- /.idea/deployment.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/.idea/deployment.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/README.md -------------------------------------------------------------------------------- /cal_ic_lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/cal_ic_lpips.py -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_16x16x16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/autoencoder/autoencoder_kl_16x16x16.yaml -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_32x32x4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/autoencoder/autoencoder_kl_32x32x4.yaml -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_64x64x3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/autoencoder/autoencoder_kl_64x64x3.yaml -------------------------------------------------------------------------------- /configs/autoencoder/autoencoder_kl_8x8x64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/autoencoder/autoencoder_kl_8x8x64.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/celebahq-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/latent-diffusion/celebahq-ldm-vq-4.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/cin-ldm-vq-f8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/latent-diffusion/cin-ldm-vq-f8.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/cin256-v2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/latent-diffusion/cin256-v2.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/ffhq-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/latent-diffusion/ffhq-ldm-vq-4.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/lsun_bedrooms-ldm-vq-4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/latent-diffusion/lsun_bedrooms-ldm-vq-4.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/lsun_churches-ldm-kl-8.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/latent-diffusion/lsun_churches-ldm-kl-8.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/txt2img-1p4B-eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/latent-diffusion/txt2img-1p4B-eval.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/txt2img-1p4B-eval_with_tokens.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/latent-diffusion/txt2img-1p4B-eval_with_tokens.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/txt2img-1p4B-finetune-encoder+embedding.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/latent-diffusion/txt2img-1p4B-finetune-encoder+embedding.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/txt2img-1p4B-finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/latent-diffusion/txt2img-1p4B-finetune.yaml -------------------------------------------------------------------------------- /configs/latent-diffusion/txt2img-1p4B-finetune_style.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/latent-diffusion/txt2img-1p4B-finetune_style.yaml -------------------------------------------------------------------------------- /configs/stable-diffusion/v1-finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/stable-diffusion/v1-finetune.yaml -------------------------------------------------------------------------------- /configs/stable-diffusion/v1-finetune_unfrozen.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/stable-diffusion/v1-finetune_unfrozen.yaml -------------------------------------------------------------------------------- /configs/stable-diffusion/v1-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/configs/stable-diffusion/v1-inference.yaml -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/environment.yaml -------------------------------------------------------------------------------- /evaluation/clip_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/evaluation/clip_eval.py -------------------------------------------------------------------------------- /generate_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/generate_mask.py -------------------------------------------------------------------------------- /generate_with_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/generate_with_mask.py -------------------------------------------------------------------------------- /ldm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/data/base.py -------------------------------------------------------------------------------- /ldm/data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/data/imagenet.py -------------------------------------------------------------------------------- /ldm/data/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/data/lsun.py -------------------------------------------------------------------------------- /ldm/data/mask_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/data/mask_generate.py -------------------------------------------------------------------------------- /ldm/data/personalized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/data/personalized.py -------------------------------------------------------------------------------- /ldm/data/personalized_style.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/data/personalized_style.py -------------------------------------------------------------------------------- /ldm/data/tmp.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/data/tmp.jpg -------------------------------------------------------------------------------- /ldm/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/lr_scheduler.py -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/diffusion/classifier.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /ldm/models/encoders.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/psp_encoder.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder.zip -------------------------------------------------------------------------------- /ldm/models/psp_encoder/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/psp_encoder/encoders/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder/encoders/helpers.py -------------------------------------------------------------------------------- /ldm/models/psp_encoder/encoders/model_irse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder/encoders/model_irse.py -------------------------------------------------------------------------------- /ldm/models/psp_encoder/encoders/psp_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder/encoders/psp_encoders.py -------------------------------------------------------------------------------- /ldm/models/psp_encoder/stylegan2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/psp_encoder/stylegan2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder/stylegan2/model.py -------------------------------------------------------------------------------- /ldm/models/psp_encoder/stylegan2/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder/stylegan2/op/__init__.py -------------------------------------------------------------------------------- /ldm/models/psp_encoder/stylegan2/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder/stylegan2/op/fused_act.py -------------------------------------------------------------------------------- /ldm/models/psp_encoder/stylegan2/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder/stylegan2/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /ldm/models/psp_encoder/stylegan2/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder/stylegan2/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /ldm/models/psp_encoder/stylegan2/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder/stylegan2/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /ldm/models/psp_encoder/stylegan2/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder/stylegan2/op/upfirdn2d.py -------------------------------------------------------------------------------- /ldm/models/psp_encoder/stylegan2/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/psp_encoder/stylegan2/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /ldm/models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/models/vit.py -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/embedding_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/embedding_manager.py -------------------------------------------------------------------------------- /ldm/modules/embedding_manager2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/embedding_manager2.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/encoders/modules_bak.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/encoders/modules_bak.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /ldm/modules/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from ldm.modules.losses.contperceptual import LPIPSWithDiscriminator -------------------------------------------------------------------------------- /ldm/modules/losses/contperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/losses/contperceptual.py -------------------------------------------------------------------------------- /ldm/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /ldm/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/modules/x_transformer.py -------------------------------------------------------------------------------- /ldm/my_model/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/my_model/models.py -------------------------------------------------------------------------------- /ldm/my_model/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/my_model/op/__init__.py -------------------------------------------------------------------------------- /ldm/my_model/op/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/my_model/op/conv2d_gradfix.py -------------------------------------------------------------------------------- /ldm/my_model/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/my_model/op/fused_act.py -------------------------------------------------------------------------------- /ldm/my_model/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/my_model/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /ldm/my_model/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/my_model/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /ldm/my_model/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/my_model/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /ldm/my_model/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/my_model/op/upfirdn2d.py -------------------------------------------------------------------------------- /ldm/my_model/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/my_model/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/ldm/util.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/main.py -------------------------------------------------------------------------------- /models/first_stage_models/kl-f16/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/first_stage_models/kl-f16/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/kl-f32/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/first_stage_models/kl-f32/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/kl-f4/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/first_stage_models/kl-f4/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/kl-f8/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/first_stage_models/kl-f8/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/vq-f16/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/first_stage_models/vq-f16/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/vq-f4-noattn/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/first_stage_models/vq-f4-noattn/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/vq-f4/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/first_stage_models/vq-f4/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/vq-f8-n256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/first_stage_models/vq-f8-n256/config.yaml -------------------------------------------------------------------------------- /models/first_stage_models/vq-f8/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/first_stage_models/vq-f8/config.yaml -------------------------------------------------------------------------------- /models/ldm/bsr_sr/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/ldm/bsr_sr/config.yaml -------------------------------------------------------------------------------- /models/ldm/celeba256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/ldm/celeba256/config.yaml -------------------------------------------------------------------------------- /models/ldm/cin256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/ldm/cin256/config.yaml -------------------------------------------------------------------------------- /models/ldm/ffhq256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/ldm/ffhq256/config.yaml -------------------------------------------------------------------------------- /models/ldm/inpainting_big/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/ldm/inpainting_big/config.yaml -------------------------------------------------------------------------------- /models/ldm/layout2img-openimages256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/ldm/layout2img-openimages256/config.yaml -------------------------------------------------------------------------------- /models/ldm/lsun_beds256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/ldm/lsun_beds256/config.yaml -------------------------------------------------------------------------------- /models/ldm/lsun_churches256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/ldm/lsun_churches256/config.yaml -------------------------------------------------------------------------------- /models/ldm/semantic_synthesis256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/ldm/semantic_synthesis256/config.yaml -------------------------------------------------------------------------------- /models/ldm/semantic_synthesis512/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/ldm/semantic_synthesis512/config.yaml -------------------------------------------------------------------------------- /models/ldm/text2img256/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/models/ldm/text2img256/config.yaml -------------------------------------------------------------------------------- /name-anomaly.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/name-anomaly.txt -------------------------------------------------------------------------------- /run-mvtec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/run-mvtec.py -------------------------------------------------------------------------------- /scripts/download_first_stages.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/scripts/download_first_stages.sh -------------------------------------------------------------------------------- /scripts/download_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/scripts/download_models.sh -------------------------------------------------------------------------------- /scripts/evaluate_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/scripts/evaluate_model.py -------------------------------------------------------------------------------- /scripts/inpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/scripts/inpaint.py -------------------------------------------------------------------------------- /scripts/latent_imagenet_diffusion.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/scripts/latent_imagenet_diffusion.ipynb -------------------------------------------------------------------------------- /scripts/sample_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/scripts/sample_diffusion.py -------------------------------------------------------------------------------- /scripts/stable_txt2img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/scripts/stable_txt2img.py -------------------------------------------------------------------------------- /scripts/txt2img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/scripts/txt2img.py -------------------------------------------------------------------------------- /taming/modules/autoencoder/lpips/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/taming/modules/autoencoder/lpips/vgg.pth -------------------------------------------------------------------------------- /test-classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/test-classification.py -------------------------------------------------------------------------------- /test-localization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/test-localization.py -------------------------------------------------------------------------------- /train-classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/train-classification.py -------------------------------------------------------------------------------- /train-localization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/train-localization.py -------------------------------------------------------------------------------- /train_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/train_mask.py -------------------------------------------------------------------------------- /unet_utils/au_pro_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/unet_utils/au_pro_util.py -------------------------------------------------------------------------------- /unet_utils/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/unet_utils/data_loader.py -------------------------------------------------------------------------------- /unet_utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/unet_utils/loss.py -------------------------------------------------------------------------------- /unet_utils/model_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/unet_utils/model_unet.py -------------------------------------------------------------------------------- /unet_utils/tensorboard_visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/unet_utils/tensorboard_visualizer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sjtuplayer/anomalydiffusion/HEAD/utils.py --------------------------------------------------------------------------------