├── .github └── workflows │ └── publish.yml ├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── assets ├── keep-imgench.jpg ├── keep-preview.png └── keep-vidench.jpg ├── modules ├── deps │ ├── wm_basicsr │ │ ├── VERSION │ │ ├── __init__.py │ │ ├── archs │ │ │ ├── __init__.py │ │ │ ├── arcface_arch.py │ │ │ ├── arch_util.py │ │ │ ├── basicvsr_arch.py │ │ │ ├── basicvsrpp_arch.py │ │ │ ├── correlation.py │ │ │ ├── dfdnet_arch.py │ │ │ ├── dfdnet_util.py │ │ │ ├── discriminator_arch.py │ │ │ ├── duf_arch.py │ │ │ ├── ecbsr_arch.py │ │ │ ├── edsr_arch.py │ │ │ ├── edvr_arch.py │ │ │ ├── gmflow │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── data │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── chairs_split.txt │ │ │ │ │ ├── datasets.py │ │ │ │ │ └── transforms.py │ │ │ │ ├── environment.yml │ │ │ │ ├── evaluate.py │ │ │ │ ├── gmflow │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── backbone.py │ │ │ │ │ ├── geometry.py │ │ │ │ │ ├── gmflow.py │ │ │ │ │ ├── matching.py │ │ │ │ │ ├── position.py │ │ │ │ │ ├── transformer.py │ │ │ │ │ ├── trident_conv.py │ │ │ │ │ └── utils.py │ │ │ │ ├── loss.py │ │ │ │ ├── main.py │ │ │ │ └── utils │ │ │ │ │ ├── dist_utils.py │ │ │ │ │ ├── flow_viz.py │ │ │ │ │ ├── frame_utils.py │ │ │ │ │ ├── logger.py │ │ │ │ │ ├── misc.py │ │ │ │ │ └── utils.py │ │ │ ├── gmflow_arch.py │ │ │ ├── hifacegan_arch.py │ │ │ ├── hifacegan_util.py │ │ │ ├── inception.py │ │ │ ├── keep_arch.py │ │ │ ├── rcan_arch.py │ │ │ ├── ridnet_arch.py │ │ │ ├── rrdbnet_arch.py │ │ │ ├── spectral_norm_arch.py │ │ │ ├── spynet_arch.py │ │ │ ├── srresnet_arch.py │ │ │ ├── srvgg_arch.py │ │ │ ├── stylegan2_arch.py │ │ │ ├── stylegan2_bilinear_arch.py │ │ │ ├── swinir_arch.py │ │ │ ├── tof_arch.py │ │ │ ├── vgg_arch.py │ │ │ └── vqgan_arch.py │ │ ├── data │ │ │ ├── __init__.py │ │ │ ├── data_sampler.py │ │ │ ├── data_util.py │ │ │ ├── degradations.py │ │ │ ├── ffhq_dataset.py │ │ │ ├── gaussian_kernels.py │ │ │ ├── meta_info │ │ │ │ ├── meta_info_DIV2K800sub_GT.txt │ │ │ │ ├── meta_info_REDS4_test_GT.txt │ │ │ │ ├── meta_info_REDS_GT.txt │ │ │ │ ├── meta_info_REDSofficial4_test_GT.txt │ │ │ │ ├── meta_info_REDSval_official_test_GT.txt │ │ │ │ ├── meta_info_Vimeo90K_test_GT.txt │ │ │ │ ├── meta_info_Vimeo90K_test_fast_GT.txt │ │ │ │ ├── meta_info_Vimeo90K_test_medium_GT.txt │ │ │ │ ├── meta_info_Vimeo90K_test_slow_GT.txt │ │ │ │ └── meta_info_Vimeo90K_train_GT.txt │ │ │ ├── paired_image_dataset.py │ │ │ ├── prefetch_dataloader.py │ │ │ ├── realesrgan_dataset.py │ │ │ ├── realesrgan_paired_dataset.py │ │ │ ├── reds_dataset.py │ │ │ ├── single_image_dataset.py │ │ │ ├── transforms.py │ │ │ ├── vfhq_real_degradation2_dataset.py │ │ │ ├── video_test_dataset.py │ │ │ └── vimeo90k_dataset.py │ │ ├── losses │ │ │ ├── __init__.py │ │ │ ├── basic_loss.py │ │ │ ├── gan_loss.py │ │ │ └── loss_util.py │ │ ├── metrics │ │ │ ├── README.md │ │ │ ├── README_CN.md │ │ │ ├── __init__.py │ │ │ ├── fid.py │ │ │ ├── metric_util.py │ │ │ ├── niqe.py │ │ │ ├── niqe_pris_params.npz │ │ │ ├── psnr_ssim.py │ │ │ └── test_metrics │ │ │ │ └── test_psnr_ssim.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── base_model.py │ │ │ ├── edvr_model.py │ │ │ ├── esrgan_model.py │ │ │ ├── hifacegan_model.py │ │ │ ├── keep_gan_model.py │ │ │ ├── keep_model.py │ │ │ ├── lr_scheduler.py │ │ │ ├── realesrgan_model.py │ │ │ ├── realesrnet_model.py │ │ │ ├── sr_model.py │ │ │ ├── srgan_model.py │ │ │ ├── stylegan2_model.py │ │ │ ├── swinir_model.py │ │ │ ├── video_base_model.py │ │ │ ├── video_gan_model.py │ │ │ ├── video_recurrent_gan_model.py │ │ │ └── video_recurrent_model.py │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── dcn │ │ │ │ ├── __init__.py │ │ │ │ ├── deform_conv.py │ │ │ │ └── src │ │ │ │ │ ├── deform_conv_cuda.cpp │ │ │ │ │ ├── deform_conv_cuda_kernel.cu │ │ │ │ │ └── deform_conv_ext.cpp │ │ │ ├── fused_act │ │ │ │ ├── __init__.py │ │ │ │ ├── fused_act.py │ │ │ │ └── src │ │ │ │ │ ├── fused_bias_act.cpp │ │ │ │ │ └── fused_bias_act_kernel.cu │ │ │ └── upfirdn2d │ │ │ │ ├── __init__.py │ │ │ │ ├── src │ │ │ │ ├── upfirdn2d.cpp │ │ │ │ └── upfirdn2d_kernel.cu │ │ │ │ └── upfirdn2d.py │ │ ├── test.py │ │ ├── train.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── color_util.py │ │ │ ├── diffjpeg.py │ │ │ ├── dist_util.py │ │ │ ├── download_util.py │ │ │ ├── file_client.py │ │ │ ├── flow_util.py │ │ │ ├── img_process_util.py │ │ │ ├── img_util.py │ │ │ ├── lmdb_util.py │ │ │ ├── logger.py │ │ │ ├── matlab_functions.py │ │ │ ├── misc.py │ │ │ ├── options.py │ │ │ ├── plot_util.py │ │ │ ├── realesrgan_utils.py │ │ │ ├── registry.py │ │ │ └── video_util.py │ │ └── version.py │ └── wm_facelib │ │ ├── detection │ │ ├── __init__.py │ │ ├── align_trans.py │ │ ├── matlab_cp2tform.py │ │ ├── retinaface │ │ │ ├── __init__.py │ │ │ ├── retinaface.py │ │ │ ├── retinaface_net.py │ │ │ └── retinaface_utils.py │ │ └── yolov5face │ │ │ ├── __init__.py │ │ │ ├── face_detector.py │ │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── common.py │ │ │ ├── experimental.py │ │ │ ├── yolo.py │ │ │ ├── yolov5l.yaml │ │ │ └── yolov5n.yaml │ │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── autoanchor.py │ │ │ ├── datasets.py │ │ │ ├── extract_ckpt.py │ │ │ ├── general.py │ │ │ └── torch_utils.py │ │ ├── parsing │ │ ├── __init__.py │ │ ├── bisenet.py │ │ ├── parsenet.py │ │ └── resnet.py │ │ └── utils │ │ ├── __init__.py │ │ ├── face_restoration_helper.py │ │ ├── face_utils.py │ │ └── misc.py ├── keep_model_loader.py ├── keep_processor.py └── utils.py ├── nodes.py ├── pyproject.toml ├── requirements.txt └── workflow-examples ├── keep-facerestore-workflow.json ├── keep-facerestore-workflow.png ├── keep-workflow.json └── keep-workflow.png /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/__init__.py -------------------------------------------------------------------------------- /assets/keep-imgench.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/assets/keep-imgench.jpg -------------------------------------------------------------------------------- /assets/keep-preview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/assets/keep-preview.png -------------------------------------------------------------------------------- /assets/keep-vidench.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/assets/keep-vidench.jpg -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/VERSION: -------------------------------------------------------------------------------- 1 | 1.4.2 2 | -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/arcface_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/arcface_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/arch_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/basicvsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/basicvsr_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/basicvsrpp_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/basicvsrpp_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/correlation.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/dfdnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/dfdnet_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/dfdnet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/dfdnet_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/discriminator_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/discriminator_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/duf_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/duf_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/ecbsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/ecbsr_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/edsr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/edsr_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/edvr_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/edvr_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/LICENSE -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/README.md -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/data/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/data/chairs_split.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/data/chairs_split.txt -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/data/datasets.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/data/transforms.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/environment.yml -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/evaluate.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/gmflow/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/gmflow/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/gmflow/backbone.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/gmflow/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/gmflow/geometry.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/gmflow/gmflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/gmflow/gmflow.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/gmflow/matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/gmflow/matching.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/gmflow/position.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/gmflow/position.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/gmflow/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/gmflow/transformer.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/gmflow/trident_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/gmflow/trident_conv.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/gmflow/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/gmflow/utils.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/loss.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/main.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/utils/dist_utils.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/utils/flow_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/utils/flow_viz.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/utils/frame_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/utils/frame_utils.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/utils/logger.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/utils/misc.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow/utils/utils.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/gmflow_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/gmflow_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/hifacegan_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/hifacegan_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/hifacegan_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/hifacegan_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/inception.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/keep_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/keep_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/rcan_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/rcan_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/ridnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/ridnet_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/rrdbnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/rrdbnet_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/spectral_norm_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/spectral_norm_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/spynet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/spynet_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/srresnet_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/srresnet_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/srvgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/srvgg_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/stylegan2_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/stylegan2_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/stylegan2_bilinear_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/stylegan2_bilinear_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/swinir_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/swinir_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/tof_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/tof_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/vgg_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/vgg_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/archs/vqgan_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/archs/vqgan_arch.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/data_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/degradations.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/ffhq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/ffhq_dataset.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/gaussian_kernels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/gaussian_kernels.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/meta_info/meta_info_REDS4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/meta_info/meta_info_REDS4_test_GT.txt -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/meta_info/meta_info_REDS_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/meta_info/meta_info_REDS_GT.txt -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/meta_info/meta_info_REDSofficial4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/meta_info/meta_info_REDSofficial4_test_GT.txt -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/meta_info/meta_info_REDSval_official_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/meta_info/meta_info_REDSval_official_test_GT.txt -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/meta_info/meta_info_Vimeo90K_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/meta_info/meta_info_Vimeo90K_test_GT.txt -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/meta_info/meta_info_Vimeo90K_test_fast_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/meta_info/meta_info_Vimeo90K_test_fast_GT.txt -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/meta_info/meta_info_Vimeo90K_test_medium_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/meta_info/meta_info_Vimeo90K_test_medium_GT.txt -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/meta_info/meta_info_Vimeo90K_test_slow_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/meta_info/meta_info_Vimeo90K_test_slow_GT.txt -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/meta_info/meta_info_Vimeo90K_train_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/meta_info/meta_info_Vimeo90K_train_GT.txt -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/realesrgan_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/realesrgan_dataset.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/realesrgan_paired_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/realesrgan_paired_dataset.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/reds_dataset.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/transforms.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/vfhq_real_degradation2_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/vfhq_real_degradation2_dataset.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/video_test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/video_test_dataset.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/data/vimeo90k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/data/vimeo90k_dataset.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/losses/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/losses/basic_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/losses/basic_loss.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/losses/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/losses/gan_loss.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/losses/loss_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/metrics/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/metrics/README.md -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/metrics/README_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/metrics/README_CN.md -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/metrics/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/metrics/fid.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/metrics/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/metrics/niqe.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/metrics/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/metrics/niqe_pris_params.npz -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/metrics/test_metrics/test_psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/metrics/test_metrics/test_psnr_ssim.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/base_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/edvr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/edvr_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/esrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/esrgan_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/hifacegan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/hifacegan_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/keep_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/keep_gan_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/keep_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/keep_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/realesrgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/realesrgan_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/realesrnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/realesrnet_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/sr_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/srgan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/srgan_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/stylegan2_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/stylegan2_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/swinir_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/swinir_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/video_base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/video_base_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/video_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/video_gan_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/video_recurrent_gan_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/video_recurrent_gan_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/models/video_recurrent_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/models/video_recurrent_model.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/dcn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/dcn/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/dcn/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/dcn/deform_conv.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/dcn/src/deform_conv_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/dcn/src/deform_conv_cuda.cpp -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/dcn/src/deform_conv_cuda_kernel.cu -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/dcn/src/deform_conv_ext.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/dcn/src/deform_conv_ext.cpp -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/fused_act/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/fused_act/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/fused_act/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/fused_act/fused_act.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/fused_act/src/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/fused_act/src/fused_bias_act.cpp -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/fused_act/src/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/fused_act/src/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/upfirdn2d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/upfirdn2d/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/upfirdn2d/src/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/upfirdn2d/src/upfirdn2d.cpp -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/upfirdn2d/src/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/ops/upfirdn2d/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/ops/upfirdn2d/upfirdn2d.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/test.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/train.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/color_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/color_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/diffjpeg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/diffjpeg.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/download_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/file_client.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/flow_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/img_process_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/img_process_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/img_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/logger.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/misc.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/options.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/plot_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/plot_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/realesrgan_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/realesrgan_utils.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/registry.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/utils/video_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/utils/video_util.py -------------------------------------------------------------------------------- /modules/deps/wm_basicsr/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_basicsr/version.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/align_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/align_trans.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/matlab_cp2tform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/matlab_cp2tform.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/retinaface/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/retinaface/retinaface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/retinaface/retinaface.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/retinaface/retinaface_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/retinaface/retinaface_net.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/retinaface/retinaface_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/retinaface/retinaface_utils.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/face_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/yolov5face/face_detector.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/yolov5face/models/common.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/yolov5face/models/experimental.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/yolov5face/models/yolo.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/models/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/yolov5face/models/yolov5l.yaml -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/models/yolov5n.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/yolov5face/models/yolov5n.yaml -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/yolov5face/utils/autoanchor.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/yolov5face/utils/datasets.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/utils/extract_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/yolov5face/utils/extract_ckpt.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/yolov5face/utils/general.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/detection/yolov5face/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/detection/yolov5face/utils/torch_utils.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/parsing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/parsing/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/parsing/bisenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/parsing/bisenet.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/parsing/parsenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/parsing/parsenet.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/parsing/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/parsing/resnet.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/utils/__init__.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/utils/face_restoration_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/utils/face_restoration_helper.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/utils/face_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/utils/face_utils.py -------------------------------------------------------------------------------- /modules/deps/wm_facelib/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/deps/wm_facelib/utils/misc.py -------------------------------------------------------------------------------- /modules/keep_model_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/keep_model_loader.py -------------------------------------------------------------------------------- /modules/keep_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/keep_processor.py -------------------------------------------------------------------------------- /modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/modules/utils.py -------------------------------------------------------------------------------- /nodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/nodes.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/requirements.txt -------------------------------------------------------------------------------- /workflow-examples/keep-facerestore-workflow.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/workflow-examples/keep-facerestore-workflow.json -------------------------------------------------------------------------------- /workflow-examples/keep-facerestore-workflow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/workflow-examples/keep-facerestore-workflow.png -------------------------------------------------------------------------------- /workflow-examples/keep-workflow.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/workflow-examples/keep-workflow.json -------------------------------------------------------------------------------- /workflow-examples/keep-workflow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildminder/ComfyUI-KEEP/HEAD/workflow-examples/keep-workflow.png --------------------------------------------------------------------------------