├── LICENSE ├── README.md ├── 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 ├── vggface2 │ ├── __init__.py │ ├── resnet.py │ ├── utils.py │ └── vggface2.py └── w_norm.py ├── datasets ├── __init__.py ├── augmentations.py ├── gt_res_dataset.py ├── images_dataset.py └── inference_dataset.py ├── licenses ├── LICENSE_HuangYG123 ├── LICENSE_S-aiueo32 ├── LICENSE_TreB1eN ├── LICENSE_lessw2020 └── LICENSE_rosinality ├── models ├── __init__.py ├── encoders │ ├── __init__.py │ ├── helpers.py │ ├── model_irse.py │ └── 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 │ ├── __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 ├── mtcnn ├── onet.npy ├── pnet.npy └── rnet.npy ├── options ├── __init__.py ├── test_options.py └── train_options.py ├── requirements.txt ├── sample ├── 0000.jpg ├── 100000.jpg ├── psp_mobile_256p.png └── train_log.jpg ├── scripts ├── align_all_parallel.py ├── calc_id_loss_parallel.py ├── calc_losses_on_images.py ├── generate_sketch_data.py ├── inference.py ├── style_mixing.py └── train.py ├── training ├── __init__.py ├── coach.py └── ranger.py └── utils ├── __init__.py ├── common.py ├── data_utils.py ├── train_utils.py └── wandb_utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/data_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/configs/data_configs.py -------------------------------------------------------------------------------- /configs/paths_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/configs/paths_config.py -------------------------------------------------------------------------------- /configs/transforms_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/configs/transforms_config.py -------------------------------------------------------------------------------- /criteria/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria/id_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/criteria/id_loss.py -------------------------------------------------------------------------------- /criteria/lpips/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria/lpips/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/criteria/lpips/lpips.py -------------------------------------------------------------------------------- /criteria/lpips/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/criteria/lpips/networks.py -------------------------------------------------------------------------------- /criteria/lpips/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/criteria/lpips/utils.py -------------------------------------------------------------------------------- /criteria/moco_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/criteria/moco_loss.py -------------------------------------------------------------------------------- /criteria/vggface2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria/vggface2/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/criteria/vggface2/resnet.py -------------------------------------------------------------------------------- /criteria/vggface2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/criteria/vggface2/utils.py -------------------------------------------------------------------------------- /criteria/vggface2/vggface2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/criteria/vggface2/vggface2.py -------------------------------------------------------------------------------- /criteria/w_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/criteria/w_norm.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/datasets/augmentations.py -------------------------------------------------------------------------------- /datasets/gt_res_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/datasets/gt_res_dataset.py -------------------------------------------------------------------------------- /datasets/images_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/datasets/images_dataset.py -------------------------------------------------------------------------------- /datasets/inference_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/datasets/inference_dataset.py -------------------------------------------------------------------------------- /licenses/LICENSE_HuangYG123: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/licenses/LICENSE_HuangYG123 -------------------------------------------------------------------------------- /licenses/LICENSE_S-aiueo32: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/licenses/LICENSE_S-aiueo32 -------------------------------------------------------------------------------- /licenses/LICENSE_TreB1eN: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/licenses/LICENSE_TreB1eN -------------------------------------------------------------------------------- /licenses/LICENSE_lessw2020: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/licenses/LICENSE_lessw2020 -------------------------------------------------------------------------------- /licenses/LICENSE_rosinality: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/licenses/LICENSE_rosinality -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/encoders/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/encoders/helpers.py -------------------------------------------------------------------------------- /models/encoders/model_irse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/encoders/model_irse.py -------------------------------------------------------------------------------- /models/encoders/psp_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/encoders/psp_encoders.py -------------------------------------------------------------------------------- /models/mtcnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/mtcnn/mtcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn_pytorch/src/__init__.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/align_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn_pytorch/src/align_trans.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn_pytorch/src/box_utils.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn_pytorch/src/detector.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/first_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn_pytorch/src/first_stage.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/get_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn_pytorch/src/get_nets.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn_pytorch/src/matlab_cp2tform.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/visualization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn_pytorch/src/visualization_utils.py -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/weights/onet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn_pytorch/src/weights/onet.npy -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/weights/pnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn_pytorch/src/weights/pnet.npy -------------------------------------------------------------------------------- /models/mtcnn/mtcnn_pytorch/src/weights/rnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/mtcnn/mtcnn_pytorch/src/weights/rnet.npy -------------------------------------------------------------------------------- /models/psp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/psp.py -------------------------------------------------------------------------------- /models/stylegan2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/stylegan2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/stylegan2/model.py -------------------------------------------------------------------------------- /models/stylegan2/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/stylegan2/op/__init__.py -------------------------------------------------------------------------------- /models/stylegan2/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/stylegan2/op/fused_act.py -------------------------------------------------------------------------------- /models/stylegan2/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/stylegan2/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /models/stylegan2/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/stylegan2/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/stylegan2/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/stylegan2/op/upfirdn2d.py -------------------------------------------------------------------------------- /models/stylegan2/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/models/stylegan2/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /mtcnn/onet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/mtcnn/onet.npy -------------------------------------------------------------------------------- /mtcnn/pnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/mtcnn/pnet.npy -------------------------------------------------------------------------------- /mtcnn/rnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/mtcnn/rnet.npy -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/options/test_options.py -------------------------------------------------------------------------------- /options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/options/train_options.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/requirements.txt -------------------------------------------------------------------------------- /sample/0000.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/sample/0000.jpg -------------------------------------------------------------------------------- /sample/100000.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/sample/100000.jpg -------------------------------------------------------------------------------- /sample/psp_mobile_256p.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/sample/psp_mobile_256p.png -------------------------------------------------------------------------------- /sample/train_log.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/sample/train_log.jpg -------------------------------------------------------------------------------- /scripts/align_all_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/scripts/align_all_parallel.py -------------------------------------------------------------------------------- /scripts/calc_id_loss_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/scripts/calc_id_loss_parallel.py -------------------------------------------------------------------------------- /scripts/calc_losses_on_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/scripts/calc_losses_on_images.py -------------------------------------------------------------------------------- /scripts/generate_sketch_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/scripts/generate_sketch_data.py -------------------------------------------------------------------------------- /scripts/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/scripts/inference.py -------------------------------------------------------------------------------- /scripts/style_mixing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/scripts/style_mixing.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/scripts/train.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /training/coach.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/training/coach.py -------------------------------------------------------------------------------- /training/ranger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/training/ranger.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/utils/common.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/utils/train_utils.py -------------------------------------------------------------------------------- /utils/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TalkUHulk/pixel2style2pixel-mobilenetv3/HEAD/utils/wandb_utils.py --------------------------------------------------------------------------------