├── .gitignore ├── LICENSE ├── README.md ├── configs ├── __init__.py ├── data_configs.py ├── paths_config.py └── transforms_config.py ├── criteria ├── .ipynb_checkpoints │ └── id_loss-checkpoint.py ├── __init__.py ├── id_loss.py ├── lpips │ ├── __init__.py │ ├── lpips.py │ ├── networks.py │ └── utils.py ├── moco_loss.py └── w_norm.py ├── demo.py ├── docs ├── custom-color-edits.png ├── prog-synthesis.png └── watercolor-synthesis.png ├── environment └── paint2pix_env.yml ├── input-images ├── 143.jpg ├── 20.jpg ├── 214.jpg ├── 262.jpg └── 823.jpg ├── licenses ├── LICENSE_S-aiueo32 ├── LICENSE_TreB1eN ├── LICENSE_alaluf ├── LICENSE_eladrich ├── LICENSE_lessw2020 ├── LICENSE_omertov └── LICENSE_rosinality ├── models ├── .ipynb_checkpoints │ └── e4e-checkpoint.py ├── __init__.py ├── e4e.py ├── e4e_modules │ ├── __init__.py │ ├── discriminator.py │ └── latent_codes_pool.py ├── encoders │ ├── .ipynb_checkpoints │ │ ├── map2style-checkpoint.py │ │ └── restyle_e4e_encoders-checkpoint.py │ ├── __init__.py │ ├── fpn_encoders.py │ ├── helpers.py │ ├── map2style.py │ ├── model_irse.py │ ├── restyle_e4e_encoders.py │ └── restyle_psp_encoders.py ├── mtcnn │ ├── __init__.py │ ├── mtcnn.py │ └── mtcnn_pytorch │ │ ├── __init__.py │ │ └── src │ │ ├── __init__.py │ │ ├── align_trans.py │ │ ├── box_utils.py │ │ ├── detector.py │ │ ├── first_stage.py │ │ ├── get_nets.py │ │ ├── matlab_cp2tform.py │ │ ├── visualization_utils.py │ │ └── weights │ │ ├── onet.npy │ │ ├── pnet.npy │ │ └── rnet.npy ├── psp.py └── stylegan2 │ ├── .ipynb_checkpoints │ └── model-checkpoint.py │ ├── __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 ├── output ├── result_0.png └── result_mask_0.png ├── predict.py └── utils ├── .ipynb_checkpoints ├── id_utils-checkpoint.py └── inference_utils-checkpoint.py ├── __init__.py ├── common.py ├── data_utils.py ├── id_utils.py ├── inference_utils.py ├── model_utils.py └── train_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/data_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/configs/data_configs.py -------------------------------------------------------------------------------- /configs/paths_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/configs/paths_config.py -------------------------------------------------------------------------------- /configs/transforms_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/configs/transforms_config.py -------------------------------------------------------------------------------- /criteria/.ipynb_checkpoints/id_loss-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/criteria/.ipynb_checkpoints/id_loss-checkpoint.py -------------------------------------------------------------------------------- /criteria/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria/id_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/criteria/id_loss.py -------------------------------------------------------------------------------- /criteria/lpips/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria/lpips/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/criteria/lpips/lpips.py -------------------------------------------------------------------------------- /criteria/lpips/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/criteria/lpips/networks.py -------------------------------------------------------------------------------- /criteria/lpips/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/criteria/lpips/utils.py -------------------------------------------------------------------------------- /criteria/moco_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/criteria/moco_loss.py -------------------------------------------------------------------------------- /criteria/w_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/criteria/w_norm.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/demo.py -------------------------------------------------------------------------------- /docs/custom-color-edits.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/docs/custom-color-edits.png -------------------------------------------------------------------------------- /docs/prog-synthesis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/docs/prog-synthesis.png -------------------------------------------------------------------------------- /docs/watercolor-synthesis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/docs/watercolor-synthesis.png -------------------------------------------------------------------------------- /environment/paint2pix_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/environment/paint2pix_env.yml -------------------------------------------------------------------------------- /input-images/143.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/input-images/143.jpg -------------------------------------------------------------------------------- /input-images/20.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/input-images/20.jpg -------------------------------------------------------------------------------- /input-images/214.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/input-images/214.jpg -------------------------------------------------------------------------------- /input-images/262.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/input-images/262.jpg -------------------------------------------------------------------------------- /input-images/823.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/input-images/823.jpg -------------------------------------------------------------------------------- /licenses/LICENSE_S-aiueo32: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/licenses/LICENSE_S-aiueo32 -------------------------------------------------------------------------------- /licenses/LICENSE_TreB1eN: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/licenses/LICENSE_TreB1eN -------------------------------------------------------------------------------- /licenses/LICENSE_alaluf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/licenses/LICENSE_alaluf -------------------------------------------------------------------------------- /licenses/LICENSE_eladrich: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/licenses/LICENSE_eladrich -------------------------------------------------------------------------------- /licenses/LICENSE_lessw2020: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/licenses/LICENSE_lessw2020 -------------------------------------------------------------------------------- /licenses/LICENSE_omertov: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/licenses/LICENSE_omertov -------------------------------------------------------------------------------- /licenses/LICENSE_rosinality: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/licenses/LICENSE_rosinality -------------------------------------------------------------------------------- /models/.ipynb_checkpoints/e4e-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/.ipynb_checkpoints/e4e-checkpoint.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/e4e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/e4e.py -------------------------------------------------------------------------------- /models/e4e_modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/e4e_modules/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/e4e_modules/discriminator.py -------------------------------------------------------------------------------- /models/e4e_modules/latent_codes_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/e4e_modules/latent_codes_pool.py -------------------------------------------------------------------------------- /models/encoders/.ipynb_checkpoints/map2style-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/encoders/.ipynb_checkpoints/map2style-checkpoint.py -------------------------------------------------------------------------------- /models/encoders/.ipynb_checkpoints/restyle_e4e_encoders-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/encoders/.ipynb_checkpoints/restyle_e4e_encoders-checkpoint.py -------------------------------------------------------------------------------- /models/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/encoders/fpn_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/encoders/fpn_encoders.py -------------------------------------------------------------------------------- /models/encoders/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/encoders/helpers.py -------------------------------------------------------------------------------- /models/encoders/map2style.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/encoders/map2style.py -------------------------------------------------------------------------------- /models/encoders/model_irse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/encoders/model_irse.py -------------------------------------------------------------------------------- /models/encoders/restyle_e4e_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/encoders/restyle_e4e_encoders.py -------------------------------------------------------------------------------- /models/encoders/restyle_psp_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/encoders/restyle_psp_encoders.py -------------------------------------------------------------------------------- /models/mtcnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/mtcnn/mtcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn_pytorch/src/__init__.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/align_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn_pytorch/src/align_trans.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn_pytorch/src/box_utils.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn_pytorch/src/detector.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/first_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn_pytorch/src/first_stage.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/get_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn_pytorch/src/get_nets.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/visualization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn_pytorch/src/visualization_utils.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/weights/onet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn_pytorch/src/weights/onet.npy -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/weights/pnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn_pytorch/src/weights/pnet.npy -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/weights/rnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/mtcnn/mtcnn_pytorch/src/weights/rnet.npy -------------------------------------------------------------------------------- /models/psp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/psp.py -------------------------------------------------------------------------------- /models/stylegan2/.ipynb_checkpoints/model-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/stylegan2/.ipynb_checkpoints/model-checkpoint.py -------------------------------------------------------------------------------- /models/stylegan2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/stylegan2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/stylegan2/model.py -------------------------------------------------------------------------------- /models/stylegan2/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/stylegan2/op/__init__.py -------------------------------------------------------------------------------- /models/stylegan2/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/stylegan2/op/fused_act.py -------------------------------------------------------------------------------- /models/stylegan2/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/stylegan2/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /models/stylegan2/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/stylegan2/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/stylegan2/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/stylegan2/op/upfirdn2d.py -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/models/stylegan2/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /output/result_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/output/result_0.png -------------------------------------------------------------------------------- /output/result_mask_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/output/result_mask_0.png -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/predict.py -------------------------------------------------------------------------------- /utils/.ipynb_checkpoints/id_utils-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/utils/.ipynb_checkpoints/id_utils-checkpoint.py -------------------------------------------------------------------------------- /utils/.ipynb_checkpoints/inference_utils-checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/utils/.ipynb_checkpoints/inference_utils-checkpoint.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/utils/common.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/id_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/utils/id_utils.py -------------------------------------------------------------------------------- /utils/inference_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/utils/inference_utils.py -------------------------------------------------------------------------------- /utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/utils/model_utils.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/1jsingh/paint2pix/HEAD/utils/train_utils.py --------------------------------------------------------------------------------