├── README.md ├── arch_inv.png ├── arch_ref.png ├── configs ├── __init__.py ├── data_configs.py ├── paths_config.py └── transforms_config.py ├── criteria ├── __init__.py ├── id_loss.py ├── lpips │ ├── __init__.py │ ├── lpips.py │ ├── networks.py │ └── utils.py └── moco_loss.py ├── datasets ├── __init__.py ├── augmentations.py ├── gt_res_dataset.py ├── images_dataset.py ├── images_dataset_label_edit.py ├── images_dataset_ref_edit.py └── inference_dataset.py ├── models ├── __init__.py ├── edit_transformer.py ├── encoders │ ├── __init__.py │ ├── edit_transformer_encoders.py │ ├── helpers.py │ ├── model_irse.py │ └── style_transformer_encoders.py ├── style_transformer.py ├── stylegan2 │ ├── __init__.py │ ├── model.py │ └── op │ │ ├── __init__.py │ │ ├── fused_act.py │ │ ├── fused_bias_act.cpp │ │ ├── fused_bias_act_kernel.cu │ │ ├── upfirdn2d.cpp │ │ ├── upfirdn2d.py │ │ └── upfirdn2d_kernel.cu └── transformer.py ├── options ├── __init__.py ├── test_options.py └── train_options.py ├── scripts ├── generate_data.py ├── inference.py ├── inference_edit.py ├── train.py ├── train_label_edit.py └── train_ref_edit.py ├── teaser.png ├── training ├── __init__.py ├── coach_invert.py ├── coach_label_edit.py ├── coach_ref_edit.py └── ranger.py └── utils ├── __init__.py ├── common.py ├── data_utils.py └── train_utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/README.md -------------------------------------------------------------------------------- /arch_inv.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/arch_inv.png -------------------------------------------------------------------------------- /arch_ref.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/arch_ref.png -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /configs/data_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/configs/data_configs.py -------------------------------------------------------------------------------- /configs/paths_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/configs/paths_config.py -------------------------------------------------------------------------------- /configs/transforms_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/configs/transforms_config.py -------------------------------------------------------------------------------- /criteria/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /criteria/id_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/criteria/id_loss.py -------------------------------------------------------------------------------- /criteria/lpips/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /criteria/lpips/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/criteria/lpips/lpips.py -------------------------------------------------------------------------------- /criteria/lpips/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/criteria/lpips/networks.py -------------------------------------------------------------------------------- /criteria/lpips/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/criteria/lpips/utils.py -------------------------------------------------------------------------------- /criteria/moco_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/criteria/moco_loss.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /datasets/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/datasets/augmentations.py -------------------------------------------------------------------------------- /datasets/gt_res_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/datasets/gt_res_dataset.py -------------------------------------------------------------------------------- /datasets/images_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/datasets/images_dataset.py -------------------------------------------------------------------------------- /datasets/images_dataset_label_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/datasets/images_dataset_label_edit.py -------------------------------------------------------------------------------- /datasets/images_dataset_ref_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/datasets/images_dataset_ref_edit.py -------------------------------------------------------------------------------- /datasets/inference_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/datasets/inference_dataset.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /models/edit_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/edit_transformer.py -------------------------------------------------------------------------------- /models/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /models/encoders/edit_transformer_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/encoders/edit_transformer_encoders.py -------------------------------------------------------------------------------- /models/encoders/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/encoders/helpers.py -------------------------------------------------------------------------------- /models/encoders/model_irse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/encoders/model_irse.py -------------------------------------------------------------------------------- /models/encoders/style_transformer_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/encoders/style_transformer_encoders.py -------------------------------------------------------------------------------- /models/style_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/style_transformer.py -------------------------------------------------------------------------------- /models/stylegan2/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /models/stylegan2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/stylegan2/model.py -------------------------------------------------------------------------------- /models/stylegan2/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/stylegan2/op/__init__.py -------------------------------------------------------------------------------- /models/stylegan2/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/stylegan2/op/fused_act.py -------------------------------------------------------------------------------- /models/stylegan2/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/stylegan2/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /models/stylegan2/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/stylegan2/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/stylegan2/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/stylegan2/op/upfirdn2d.py -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/stylegan2/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/models/transformer.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/options/test_options.py -------------------------------------------------------------------------------- /options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/options/train_options.py -------------------------------------------------------------------------------- /scripts/generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/scripts/generate_data.py -------------------------------------------------------------------------------- /scripts/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/scripts/inference.py -------------------------------------------------------------------------------- /scripts/inference_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/scripts/inference_edit.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/scripts/train.py -------------------------------------------------------------------------------- /scripts/train_label_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/scripts/train_label_edit.py -------------------------------------------------------------------------------- /scripts/train_ref_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/scripts/train_ref_edit.py -------------------------------------------------------------------------------- /teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/teaser.png -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /training/coach_invert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/training/coach_invert.py -------------------------------------------------------------------------------- /training/coach_label_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/training/coach_label_edit.py -------------------------------------------------------------------------------- /training/coach_ref_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/training/coach_ref_edit.py -------------------------------------------------------------------------------- /training/ranger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/training/ranger.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/utils/common.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sapphire497/style-transformer/HEAD/utils/train_utils.py --------------------------------------------------------------------------------