├── .gitattributes ├── .github └── workflows │ └── publish_action.yml ├── CHANGELOG.md ├── README.md ├── __init__.py ├── gpeno ├── .gitignore ├── README.md ├── __init_paths.py ├── align_faces.py ├── demo.py ├── distributed.py ├── face_colorization.py ├── face_detect │ ├── .DS_Store │ ├── data │ │ ├── FDDB │ │ │ └── img_list.txt │ │ ├── __init__.py │ │ ├── config.py │ │ ├── data_augment.py │ │ └── wider_face.py │ ├── facemodels │ │ ├── __init__.py │ │ ├── net.py │ │ └── retinaface.py │ ├── layers │ │ ├── __init__.py │ │ ├── functions │ │ │ └── prior_box.py │ │ └── modules │ │ │ ├── __init__.py │ │ │ └── multibox_loss.py │ ├── retinaface_detection.py │ └── utils │ │ ├── __init__.py │ │ ├── box_utils.py │ │ ├── nms │ │ ├── __init__.py │ │ └── py_cpu_nms.py │ │ └── timer.py ├── face_enhancement.py ├── face_inpainting.py ├── face_model │ ├── face_gan.py │ ├── gpen_model.py │ └── op │ │ ├── __init__.py │ │ ├── fused_act.py │ │ ├── fused_bias_act.cpp │ │ ├── fused_bias_act_kernel.cu │ │ ├── upfirdn2d.cpp │ │ ├── upfirdn2d.py │ │ └── upfirdn2d_kernel.cu ├── face_parse │ ├── blocks.py │ ├── face_parsing.py │ ├── face_parsing_broken.py │ ├── mask.png │ ├── parse_model.py │ └── test.png ├── misc │ ├── cog.yaml │ ├── onnx_export.py │ └── predict.py ├── requirements.txt └── training │ ├── data_loader │ ├── dataset_face.py │ └── degradations.py │ ├── loss │ ├── helpers.py │ ├── id_loss.py │ └── model_irse.py │ └── lpips │ ├── __init__.py │ ├── lpips.py │ ├── pretrained_networks.py │ ├── trainer.py │ └── weights │ ├── v0.0 │ ├── alex.pth │ ├── squeeze.pth │ └── vgg.pth │ └── v0.1 │ ├── alex.pth │ ├── squeeze.pth │ └── vgg.pth ├── pyproject.toml └── workflows └── workflow_gpeno.png /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/workflows/publish_action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/.github/workflows/publish_action.yml -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/__init__.py -------------------------------------------------------------------------------- /gpeno/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/.gitignore -------------------------------------------------------------------------------- /gpeno/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/README.md -------------------------------------------------------------------------------- /gpeno/__init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/__init_paths.py -------------------------------------------------------------------------------- /gpeno/align_faces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/align_faces.py -------------------------------------------------------------------------------- /gpeno/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/demo.py -------------------------------------------------------------------------------- /gpeno/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/distributed.py -------------------------------------------------------------------------------- /gpeno/face_colorization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_colorization.py -------------------------------------------------------------------------------- /gpeno/face_detect/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/.DS_Store -------------------------------------------------------------------------------- /gpeno/face_detect/data/FDDB/img_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/data/FDDB/img_list.txt -------------------------------------------------------------------------------- /gpeno/face_detect/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/data/__init__.py -------------------------------------------------------------------------------- /gpeno/face_detect/data/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/data/config.py -------------------------------------------------------------------------------- /gpeno/face_detect/data/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/data/data_augment.py -------------------------------------------------------------------------------- /gpeno/face_detect/data/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/data/wider_face.py -------------------------------------------------------------------------------- /gpeno/face_detect/facemodels/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpeno/face_detect/facemodels/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/facemodels/net.py -------------------------------------------------------------------------------- /gpeno/face_detect/facemodels/retinaface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/facemodels/retinaface.py -------------------------------------------------------------------------------- /gpeno/face_detect/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/layers/__init__.py -------------------------------------------------------------------------------- /gpeno/face_detect/layers/functions/prior_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/layers/functions/prior_box.py -------------------------------------------------------------------------------- /gpeno/face_detect/layers/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/layers/modules/__init__.py -------------------------------------------------------------------------------- /gpeno/face_detect/layers/modules/multibox_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/layers/modules/multibox_loss.py -------------------------------------------------------------------------------- /gpeno/face_detect/retinaface_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/retinaface_detection.py -------------------------------------------------------------------------------- /gpeno/face_detect/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpeno/face_detect/utils/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/utils/box_utils.py -------------------------------------------------------------------------------- /gpeno/face_detect/utils/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gpeno/face_detect/utils/nms/py_cpu_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/utils/nms/py_cpu_nms.py -------------------------------------------------------------------------------- /gpeno/face_detect/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_detect/utils/timer.py -------------------------------------------------------------------------------- /gpeno/face_enhancement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_enhancement.py -------------------------------------------------------------------------------- /gpeno/face_inpainting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_inpainting.py -------------------------------------------------------------------------------- /gpeno/face_model/face_gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_model/face_gan.py -------------------------------------------------------------------------------- /gpeno/face_model/gpen_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_model/gpen_model.py -------------------------------------------------------------------------------- /gpeno/face_model/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_model/op/__init__.py -------------------------------------------------------------------------------- /gpeno/face_model/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_model/op/fused_act.py -------------------------------------------------------------------------------- /gpeno/face_model/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_model/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /gpeno/face_model/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_model/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /gpeno/face_model/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_model/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /gpeno/face_model/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_model/op/upfirdn2d.py -------------------------------------------------------------------------------- /gpeno/face_model/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_model/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /gpeno/face_parse/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_parse/blocks.py -------------------------------------------------------------------------------- /gpeno/face_parse/face_parsing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_parse/face_parsing.py -------------------------------------------------------------------------------- /gpeno/face_parse/face_parsing_broken.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_parse/face_parsing_broken.py -------------------------------------------------------------------------------- /gpeno/face_parse/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_parse/mask.png -------------------------------------------------------------------------------- /gpeno/face_parse/parse_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_parse/parse_model.py -------------------------------------------------------------------------------- /gpeno/face_parse/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/face_parse/test.png -------------------------------------------------------------------------------- /gpeno/misc/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/misc/cog.yaml -------------------------------------------------------------------------------- /gpeno/misc/onnx_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/misc/onnx_export.py -------------------------------------------------------------------------------- /gpeno/misc/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/misc/predict.py -------------------------------------------------------------------------------- /gpeno/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/requirements.txt -------------------------------------------------------------------------------- /gpeno/training/data_loader/dataset_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/data_loader/dataset_face.py -------------------------------------------------------------------------------- /gpeno/training/data_loader/degradations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/data_loader/degradations.py -------------------------------------------------------------------------------- /gpeno/training/loss/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/loss/helpers.py -------------------------------------------------------------------------------- /gpeno/training/loss/id_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/loss/id_loss.py -------------------------------------------------------------------------------- /gpeno/training/loss/model_irse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/loss/model_irse.py -------------------------------------------------------------------------------- /gpeno/training/lpips/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/lpips/__init__.py -------------------------------------------------------------------------------- /gpeno/training/lpips/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/lpips/lpips.py -------------------------------------------------------------------------------- /gpeno/training/lpips/pretrained_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/lpips/pretrained_networks.py -------------------------------------------------------------------------------- /gpeno/training/lpips/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/lpips/trainer.py -------------------------------------------------------------------------------- /gpeno/training/lpips/weights/v0.0/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/lpips/weights/v0.0/alex.pth -------------------------------------------------------------------------------- /gpeno/training/lpips/weights/v0.0/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/lpips/weights/v0.0/squeeze.pth -------------------------------------------------------------------------------- /gpeno/training/lpips/weights/v0.0/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/lpips/weights/v0.0/vgg.pth -------------------------------------------------------------------------------- /gpeno/training/lpips/weights/v0.1/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/lpips/weights/v0.1/alex.pth -------------------------------------------------------------------------------- /gpeno/training/lpips/weights/v0.1/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/lpips/weights/v0.1/squeeze.pth -------------------------------------------------------------------------------- /gpeno/training/lpips/weights/v0.1/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/gpeno/training/lpips/weights/v0.1/vgg.pth -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/pyproject.toml -------------------------------------------------------------------------------- /workflows/workflow_gpeno.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SparknightLLC/ComfyUI-GPENO/HEAD/workflows/workflow_gpeno.png --------------------------------------------------------------------------------