├── .gitignore ├── LICENSE ├── README.md ├── ckpts └── .gitkeep ├── config ├── experiments │ ├── age_debias_1_ffhq.cfg │ ├── age_debias_1_metfaces.cfg │ ├── age_debias_2_ffhq.cfg │ ├── age_debias_2_metfaces.cfg │ ├── class_blond_hair.cfg │ ├── class_eyeglasses.cfg │ ├── class_eyeglasses_old.cfg │ ├── class_eyeglasses_young.cfg │ ├── class_female.cfg │ ├── class_female_glasses.cfg │ ├── class_female_noglasses.cfg │ ├── class_female_old.cfg │ ├── class_female_young.cfg │ ├── class_male.cfg │ ├── class_male_glasses.cfg │ ├── class_male_noglasses.cfg │ ├── class_male_old.cfg │ ├── class_male_young.cfg │ ├── class_noblond_hair.cfg │ ├── class_noeyeglasses.cfg │ ├── class_noeyeglasses_old.cfg │ ├── class_noeyeglasses_young.cfg │ ├── clip_a_baby_ffhq.cfg │ ├── clip_a_baby_ffhq1024_stylenerf.cfg │ ├── clip_a_baby_ffhq256_diffae.cfg │ ├── clip_a_boy_ffhq.cfg │ ├── clip_a_girl_ffhq.cfg │ ├── clip_a_person_without_makeup.cfg │ ├── clip_an_asian_man_ffhq.cfg │ ├── clip_autumn_scene_landscape.cfg │ ├── clip_ink_wash_of_a_church_near_forest_under_moonlight_biggan.cfg │ ├── clip_painting_of_a_melancholy_robot_biggan.cfg │ ├── clip_photo_of_a_glow_and_light_dog_biggan.cfg │ ├── clip_winter_scene_landscape.cfg │ ├── debias_ebm_ffhq.cfg │ ├── debias_ebm_metfaces.cfg │ ├── debias_ebm_person_without_makeup.cfg │ ├── gender_debias_1_metfaces.cfg │ ├── gender_debias_2_metfaces.cfg │ ├── gender_debias_2_person_without_makeup.cfg │ ├── inverse_graphics_pose_ffhq.cfg │ ├── inverse_graphics_pose_ffhq_test_fid.cfg │ ├── no_debias_person_without_makeup.cfg │ ├── race_debias_1_ffhq.cfg │ ├── race_debias_1_metfaces.cfg │ ├── race_debias_2_ffhq.cfg │ └── race_debias_2_metfaces.cfg └── tasks │ ├── class.cfg │ ├── clip.cfg │ ├── debias_age.cfg │ ├── debias_gender.cfg │ ├── debias_race.cfg │ ├── get_debias_ebm.cfg │ ├── pose.cfg │ └── pose_fid.cfg ├── docs ├── first_fig.png ├── overview.png ├── pipeline_overview.png ├── poster.pdf └── poster.png ├── environment.yml ├── evaluation ├── __init__.py ├── class.py ├── debias.py ├── example.py ├── ffhq_fid_pose.py ├── get_debias_ebm.py └── multi_task.py ├── main.py ├── model ├── __init__.py ├── debias_ebm.py ├── energy │ ├── __init__.py │ ├── class_condition.py │ ├── clip_guide.py │ ├── debias.py │ ├── get_energy.py │ ├── id.py │ ├── pose.py │ ├── prior_y.py │ └── prior_z.py ├── gan_wrapper │ ├── __init__.py │ ├── biggan_wrapper.py │ ├── diffae_wrapper.py │ ├── get_gan_wrapper.py │ ├── promptgan2_wrapper.py │ ├── stylegan2_wrapper.py │ └── stylenerf_wrapper.py ├── lib │ ├── __init__.py │ ├── biggan │ │ ├── LICENSE │ │ ├── __init__.py │ │ ├── config.py │ │ ├── file_utils.py │ │ ├── model.py │ │ └── utils.py │ ├── celeba │ │ ├── __init__.py │ │ └── classifier.py │ ├── decalib │ │ ├── __init__.py │ │ ├── data │ │ │ ├── fixed_displacement_256.npy │ │ │ ├── head_template.obj │ │ │ ├── landmark_embedding.npy │ │ │ ├── mean_texture.jpg │ │ │ ├── texture_data_256.npy │ │ │ ├── uv_face_eye_mask.png │ │ │ └── uv_face_mask.png │ │ ├── datasets │ │ │ ├── aflw2000.py │ │ │ ├── build_datasets.py │ │ │ ├── datasets.py │ │ │ ├── detectors.py │ │ │ ├── ethnicity.py │ │ │ ├── now.py │ │ │ └── train_datasets.py │ │ ├── deca.py │ │ ├── models │ │ │ ├── FLAME.py │ │ │ ├── decoders.py │ │ │ ├── encoders.py │ │ │ ├── frnet.py │ │ │ ├── lbs.py │ │ │ └── resnet.py │ │ └── utils │ │ │ ├── config.py │ │ │ ├── lossfunc.py │ │ │ ├── renderer.py │ │ │ ├── rotation_converter.py │ │ │ ├── tensor_cropper.py │ │ │ └── util.py │ ├── diffae │ │ ├── choices.py │ │ ├── config.py │ │ ├── config_base.py │ │ ├── diffusion │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── diffusion.py │ │ │ └── resample.py │ │ ├── dist_utils.py │ │ ├── experiment.py │ │ ├── model │ │ │ ├── __init__.py │ │ │ ├── blocks.py │ │ │ ├── latentnet.py │ │ │ ├── nn.py │ │ │ ├── unet.py │ │ │ └── unet_autoenc.py │ │ ├── renderer.py │ │ ├── templates.py │ │ └── templates_latent.py │ ├── diffaug │ │ ├── DiffAugment_pytorch.py │ │ └── __init__.py │ ├── fairface │ │ ├── __init__.py │ │ └── classifier.py │ ├── id_recognition │ │ ├── __init__.py │ │ ├── helpers.py │ │ └── model_irse.py │ ├── stylegan2 │ │ ├── __init__.py │ │ ├── op │ │ │ ├── __init__.py │ │ │ ├── conv2d_gradfix.py │ │ │ ├── fused_act.py │ │ │ ├── fused_bias_act.cpp │ │ │ ├── fused_bias_act_kernel.cu │ │ │ ├── upfirdn2d.cpp │ │ │ ├── upfirdn2d.py │ │ │ └── upfirdn2d_kernel.cu │ │ └── sg2_model.py │ └── stylenerf │ │ ├── LICENSE.txt │ │ ├── README.md │ │ ├── dnnlib │ │ ├── __init__.py │ │ ├── camera.py │ │ ├── filters.py │ │ ├── geometry.py │ │ └── util.py │ │ ├── generate.py │ │ ├── launcher.py │ │ ├── legacy.py │ │ ├── renderer.py │ │ ├── requirements.txt │ │ ├── run_train.py │ │ ├── torch_utils │ │ ├── __init__.py │ │ ├── custom_ops.py │ │ ├── distributed_utils.py │ │ ├── misc.py │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── bias_act.cpp │ │ │ ├── bias_act.cu │ │ │ ├── bias_act.h │ │ │ ├── bias_act.py │ │ │ ├── conv2d_gradfix.py │ │ │ ├── conv2d_resample.py │ │ │ ├── filtered_lrelu.cpp │ │ │ ├── filtered_lrelu.cu │ │ │ ├── filtered_lrelu.h │ │ │ ├── filtered_lrelu.py │ │ │ ├── filtered_lrelu_ns.cu │ │ │ ├── filtered_lrelu_rd.cu │ │ │ ├── filtered_lrelu_wr.cu │ │ │ ├── fma.py │ │ │ ├── grid_sample_gradfix.py │ │ │ ├── hash_sample.cpp │ │ │ ├── hash_sample.cu │ │ │ ├── hash_sample.h │ │ │ ├── hash_sample.py │ │ │ ├── nerf_utils.cu │ │ │ ├── nerf_utils.py │ │ │ ├── upfirdn2d.cpp │ │ │ ├── upfirdn2d.cu │ │ │ ├── upfirdn2d.h │ │ │ ├── upfirdn2d.py │ │ │ └── utils.h │ │ ├── persistence.py │ │ └── training_stats.py │ │ ├── training │ │ ├── __init__.py │ │ ├── augment.py │ │ ├── data_utils.py │ │ ├── dataset.py │ │ ├── facial_recognition │ │ │ ├── __init__.py │ │ │ ├── helpers.py │ │ │ └── model_irse.py │ │ ├── loss.py │ │ ├── networks.py │ │ ├── stylenerf.py │ │ └── training_loop.py │ │ ├── visualizer.py │ │ └── viz │ │ ├── __init__.py │ │ ├── camera_widget.py │ │ ├── capture_widget.py │ │ ├── equivariance_widget.py │ │ ├── latent_widget.py │ │ ├── layer_widget.py │ │ ├── performance_widget.py │ │ ├── pickle_widget.py │ │ ├── renderer.py │ │ ├── stylemix_widget.py │ │ └── trunc_noise_widget.py ├── mapping.py ├── model_utils.py ├── prompt_gan.py ├── prompt_gan_base.py ├── prompt_gan_pose.py └── vanilla_gan.py ├── preprocess ├── __init__.py ├── empty.py ├── empty_small_eval.py ├── pose.py └── to_model.py ├── raw_data ├── __init__.py └── empty.py ├── trainer ├── __init__.py └── trainer.py ├── utils ├── __init__.py ├── config_utils.py ├── file_utils.py ├── program_utils.py └── training_arguments.py └── visualization ├── __init__.py ├── empty.py ├── pose.py └── single_image.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/README.md -------------------------------------------------------------------------------- /ckpts/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/experiments/age_debias_1_ffhq.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/age_debias_1_ffhq.cfg -------------------------------------------------------------------------------- /config/experiments/age_debias_1_metfaces.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/age_debias_1_metfaces.cfg -------------------------------------------------------------------------------- /config/experiments/age_debias_2_ffhq.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/age_debias_2_ffhq.cfg -------------------------------------------------------------------------------- /config/experiments/age_debias_2_metfaces.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/age_debias_2_metfaces.cfg -------------------------------------------------------------------------------- /config/experiments/class_blond_hair.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_blond_hair.cfg -------------------------------------------------------------------------------- /config/experiments/class_eyeglasses.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_eyeglasses.cfg -------------------------------------------------------------------------------- /config/experiments/class_eyeglasses_old.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_eyeglasses_old.cfg -------------------------------------------------------------------------------- /config/experiments/class_eyeglasses_young.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_eyeglasses_young.cfg -------------------------------------------------------------------------------- /config/experiments/class_female.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_female.cfg -------------------------------------------------------------------------------- /config/experiments/class_female_glasses.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_female_glasses.cfg -------------------------------------------------------------------------------- /config/experiments/class_female_noglasses.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_female_noglasses.cfg -------------------------------------------------------------------------------- /config/experiments/class_female_old.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_female_old.cfg -------------------------------------------------------------------------------- /config/experiments/class_female_young.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_female_young.cfg -------------------------------------------------------------------------------- /config/experiments/class_male.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_male.cfg -------------------------------------------------------------------------------- /config/experiments/class_male_glasses.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_male_glasses.cfg -------------------------------------------------------------------------------- /config/experiments/class_male_noglasses.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_male_noglasses.cfg -------------------------------------------------------------------------------- /config/experiments/class_male_old.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_male_old.cfg -------------------------------------------------------------------------------- /config/experiments/class_male_young.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_male_young.cfg -------------------------------------------------------------------------------- /config/experiments/class_noblond_hair.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_noblond_hair.cfg -------------------------------------------------------------------------------- /config/experiments/class_noeyeglasses.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_noeyeglasses.cfg -------------------------------------------------------------------------------- /config/experiments/class_noeyeglasses_old.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_noeyeglasses_old.cfg -------------------------------------------------------------------------------- /config/experiments/class_noeyeglasses_young.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/class_noeyeglasses_young.cfg -------------------------------------------------------------------------------- /config/experiments/clip_a_baby_ffhq.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_a_baby_ffhq.cfg -------------------------------------------------------------------------------- /config/experiments/clip_a_baby_ffhq1024_stylenerf.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_a_baby_ffhq1024_stylenerf.cfg -------------------------------------------------------------------------------- /config/experiments/clip_a_baby_ffhq256_diffae.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_a_baby_ffhq256_diffae.cfg -------------------------------------------------------------------------------- /config/experiments/clip_a_boy_ffhq.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_a_boy_ffhq.cfg -------------------------------------------------------------------------------- /config/experiments/clip_a_girl_ffhq.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_a_girl_ffhq.cfg -------------------------------------------------------------------------------- /config/experiments/clip_a_person_without_makeup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_a_person_without_makeup.cfg -------------------------------------------------------------------------------- /config/experiments/clip_an_asian_man_ffhq.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_an_asian_man_ffhq.cfg -------------------------------------------------------------------------------- /config/experiments/clip_autumn_scene_landscape.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_autumn_scene_landscape.cfg -------------------------------------------------------------------------------- /config/experiments/clip_ink_wash_of_a_church_near_forest_under_moonlight_biggan.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_ink_wash_of_a_church_near_forest_under_moonlight_biggan.cfg -------------------------------------------------------------------------------- /config/experiments/clip_painting_of_a_melancholy_robot_biggan.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_painting_of_a_melancholy_robot_biggan.cfg -------------------------------------------------------------------------------- /config/experiments/clip_photo_of_a_glow_and_light_dog_biggan.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_photo_of_a_glow_and_light_dog_biggan.cfg -------------------------------------------------------------------------------- /config/experiments/clip_winter_scene_landscape.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/clip_winter_scene_landscape.cfg -------------------------------------------------------------------------------- /config/experiments/debias_ebm_ffhq.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/debias_ebm_ffhq.cfg -------------------------------------------------------------------------------- /config/experiments/debias_ebm_metfaces.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/debias_ebm_metfaces.cfg -------------------------------------------------------------------------------- /config/experiments/debias_ebm_person_without_makeup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/debias_ebm_person_without_makeup.cfg -------------------------------------------------------------------------------- /config/experiments/gender_debias_1_metfaces.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/gender_debias_1_metfaces.cfg -------------------------------------------------------------------------------- /config/experiments/gender_debias_2_metfaces.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/gender_debias_2_metfaces.cfg -------------------------------------------------------------------------------- /config/experiments/gender_debias_2_person_without_makeup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/gender_debias_2_person_without_makeup.cfg -------------------------------------------------------------------------------- /config/experiments/inverse_graphics_pose_ffhq.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/inverse_graphics_pose_ffhq.cfg -------------------------------------------------------------------------------- /config/experiments/inverse_graphics_pose_ffhq_test_fid.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/inverse_graphics_pose_ffhq_test_fid.cfg -------------------------------------------------------------------------------- /config/experiments/no_debias_person_without_makeup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/no_debias_person_without_makeup.cfg -------------------------------------------------------------------------------- /config/experiments/race_debias_1_ffhq.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/race_debias_1_ffhq.cfg -------------------------------------------------------------------------------- /config/experiments/race_debias_1_metfaces.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/race_debias_1_metfaces.cfg -------------------------------------------------------------------------------- /config/experiments/race_debias_2_ffhq.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/race_debias_2_ffhq.cfg -------------------------------------------------------------------------------- /config/experiments/race_debias_2_metfaces.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/experiments/race_debias_2_metfaces.cfg -------------------------------------------------------------------------------- /config/tasks/class.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/tasks/class.cfg -------------------------------------------------------------------------------- /config/tasks/clip.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/tasks/clip.cfg -------------------------------------------------------------------------------- /config/tasks/debias_age.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/tasks/debias_age.cfg -------------------------------------------------------------------------------- /config/tasks/debias_gender.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/tasks/debias_gender.cfg -------------------------------------------------------------------------------- /config/tasks/debias_race.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/tasks/debias_race.cfg -------------------------------------------------------------------------------- /config/tasks/get_debias_ebm.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/tasks/get_debias_ebm.cfg -------------------------------------------------------------------------------- /config/tasks/pose.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/tasks/pose.cfg -------------------------------------------------------------------------------- /config/tasks/pose_fid.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/config/tasks/pose_fid.cfg -------------------------------------------------------------------------------- /docs/first_fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/docs/first_fig.png -------------------------------------------------------------------------------- /docs/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/docs/overview.png -------------------------------------------------------------------------------- /docs/pipeline_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/docs/pipeline_overview.png -------------------------------------------------------------------------------- /docs/poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/docs/poster.pdf -------------------------------------------------------------------------------- /docs/poster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/docs/poster.png -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/environment.yml -------------------------------------------------------------------------------- /evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /evaluation/class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/evaluation/class.py -------------------------------------------------------------------------------- /evaluation/debias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/evaluation/debias.py -------------------------------------------------------------------------------- /evaluation/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/evaluation/example.py -------------------------------------------------------------------------------- /evaluation/ffhq_fid_pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/evaluation/ffhq_fid_pose.py -------------------------------------------------------------------------------- /evaluation/get_debias_ebm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/evaluation/get_debias_ebm.py -------------------------------------------------------------------------------- /evaluation/multi_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/evaluation/multi_task.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/main.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /model/debias_ebm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/debias_ebm.py -------------------------------------------------------------------------------- /model/energy/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /model/energy/class_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/energy/class_condition.py -------------------------------------------------------------------------------- /model/energy/clip_guide.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/energy/clip_guide.py -------------------------------------------------------------------------------- /model/energy/debias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/energy/debias.py -------------------------------------------------------------------------------- /model/energy/get_energy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/energy/get_energy.py -------------------------------------------------------------------------------- /model/energy/id.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/energy/id.py -------------------------------------------------------------------------------- /model/energy/pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/energy/pose.py -------------------------------------------------------------------------------- /model/energy/prior_y.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/energy/prior_y.py -------------------------------------------------------------------------------- /model/energy/prior_z.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/energy/prior_z.py -------------------------------------------------------------------------------- /model/gan_wrapper/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /model/gan_wrapper/biggan_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/gan_wrapper/biggan_wrapper.py -------------------------------------------------------------------------------- /model/gan_wrapper/diffae_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/gan_wrapper/diffae_wrapper.py -------------------------------------------------------------------------------- /model/gan_wrapper/get_gan_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/gan_wrapper/get_gan_wrapper.py -------------------------------------------------------------------------------- /model/gan_wrapper/promptgan2_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/gan_wrapper/promptgan2_wrapper.py -------------------------------------------------------------------------------- /model/gan_wrapper/stylegan2_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/gan_wrapper/stylegan2_wrapper.py -------------------------------------------------------------------------------- /model/gan_wrapper/stylenerf_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/gan_wrapper/stylenerf_wrapper.py -------------------------------------------------------------------------------- /model/lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/__init__.py -------------------------------------------------------------------------------- /model/lib/biggan/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/biggan/LICENSE -------------------------------------------------------------------------------- /model/lib/biggan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/biggan/__init__.py -------------------------------------------------------------------------------- /model/lib/biggan/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/biggan/config.py -------------------------------------------------------------------------------- /model/lib/biggan/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/biggan/file_utils.py -------------------------------------------------------------------------------- /model/lib/biggan/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/biggan/model.py -------------------------------------------------------------------------------- /model/lib/biggan/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/biggan/utils.py -------------------------------------------------------------------------------- /model/lib/celeba/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /model/lib/celeba/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/celeba/classifier.py -------------------------------------------------------------------------------- /model/lib/decalib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/lib/decalib/data/fixed_displacement_256.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/data/fixed_displacement_256.npy -------------------------------------------------------------------------------- /model/lib/decalib/data/head_template.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/data/head_template.obj -------------------------------------------------------------------------------- /model/lib/decalib/data/landmark_embedding.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/data/landmark_embedding.npy -------------------------------------------------------------------------------- /model/lib/decalib/data/mean_texture.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/data/mean_texture.jpg -------------------------------------------------------------------------------- /model/lib/decalib/data/texture_data_256.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/data/texture_data_256.npy -------------------------------------------------------------------------------- /model/lib/decalib/data/uv_face_eye_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/data/uv_face_eye_mask.png -------------------------------------------------------------------------------- /model/lib/decalib/data/uv_face_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/data/uv_face_mask.png -------------------------------------------------------------------------------- /model/lib/decalib/datasets/aflw2000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/datasets/aflw2000.py -------------------------------------------------------------------------------- /model/lib/decalib/datasets/build_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/datasets/build_datasets.py -------------------------------------------------------------------------------- /model/lib/decalib/datasets/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/datasets/datasets.py -------------------------------------------------------------------------------- /model/lib/decalib/datasets/detectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/datasets/detectors.py -------------------------------------------------------------------------------- /model/lib/decalib/datasets/ethnicity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/datasets/ethnicity.py -------------------------------------------------------------------------------- /model/lib/decalib/datasets/now.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/datasets/now.py -------------------------------------------------------------------------------- /model/lib/decalib/datasets/train_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/datasets/train_datasets.py -------------------------------------------------------------------------------- /model/lib/decalib/deca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/deca.py -------------------------------------------------------------------------------- /model/lib/decalib/models/FLAME.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/models/FLAME.py -------------------------------------------------------------------------------- /model/lib/decalib/models/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/models/decoders.py -------------------------------------------------------------------------------- /model/lib/decalib/models/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/models/encoders.py -------------------------------------------------------------------------------- /model/lib/decalib/models/frnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/models/frnet.py -------------------------------------------------------------------------------- /model/lib/decalib/models/lbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/models/lbs.py -------------------------------------------------------------------------------- /model/lib/decalib/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/models/resnet.py -------------------------------------------------------------------------------- /model/lib/decalib/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/utils/config.py -------------------------------------------------------------------------------- /model/lib/decalib/utils/lossfunc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/utils/lossfunc.py -------------------------------------------------------------------------------- /model/lib/decalib/utils/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/utils/renderer.py -------------------------------------------------------------------------------- /model/lib/decalib/utils/rotation_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/utils/rotation_converter.py -------------------------------------------------------------------------------- /model/lib/decalib/utils/tensor_cropper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/utils/tensor_cropper.py -------------------------------------------------------------------------------- /model/lib/decalib/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/decalib/utils/util.py -------------------------------------------------------------------------------- /model/lib/diffae/choices.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/choices.py -------------------------------------------------------------------------------- /model/lib/diffae/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/config.py -------------------------------------------------------------------------------- /model/lib/diffae/config_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/config_base.py -------------------------------------------------------------------------------- /model/lib/diffae/diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/diffusion/__init__.py -------------------------------------------------------------------------------- /model/lib/diffae/diffusion/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/diffusion/base.py -------------------------------------------------------------------------------- /model/lib/diffae/diffusion/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/diffusion/diffusion.py -------------------------------------------------------------------------------- /model/lib/diffae/diffusion/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/diffusion/resample.py -------------------------------------------------------------------------------- /model/lib/diffae/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/dist_utils.py -------------------------------------------------------------------------------- /model/lib/diffae/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/experiment.py -------------------------------------------------------------------------------- /model/lib/diffae/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/model/__init__.py -------------------------------------------------------------------------------- /model/lib/diffae/model/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/model/blocks.py -------------------------------------------------------------------------------- /model/lib/diffae/model/latentnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/model/latentnet.py -------------------------------------------------------------------------------- /model/lib/diffae/model/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/model/nn.py -------------------------------------------------------------------------------- /model/lib/diffae/model/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/model/unet.py -------------------------------------------------------------------------------- /model/lib/diffae/model/unet_autoenc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/model/unet_autoenc.py -------------------------------------------------------------------------------- /model/lib/diffae/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/renderer.py -------------------------------------------------------------------------------- /model/lib/diffae/templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/templates.py -------------------------------------------------------------------------------- /model/lib/diffae/templates_latent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffae/templates_latent.py -------------------------------------------------------------------------------- /model/lib/diffaug/DiffAugment_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/diffaug/DiffAugment_pytorch.py -------------------------------------------------------------------------------- /model/lib/diffaug/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /model/lib/fairface/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /model/lib/fairface/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/fairface/classifier.py -------------------------------------------------------------------------------- /model/lib/id_recognition/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /model/lib/id_recognition/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/id_recognition/helpers.py -------------------------------------------------------------------------------- /model/lib/id_recognition/model_irse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/id_recognition/model_irse.py -------------------------------------------------------------------------------- /model/lib/stylegan2/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /model/lib/stylegan2/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylegan2/op/__init__.py -------------------------------------------------------------------------------- /model/lib/stylegan2/op/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylegan2/op/conv2d_gradfix.py -------------------------------------------------------------------------------- /model/lib/stylegan2/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylegan2/op/fused_act.py -------------------------------------------------------------------------------- /model/lib/stylegan2/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylegan2/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /model/lib/stylegan2/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylegan2/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /model/lib/stylegan2/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylegan2/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /model/lib/stylegan2/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylegan2/op/upfirdn2d.py -------------------------------------------------------------------------------- /model/lib/stylegan2/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylegan2/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /model/lib/stylegan2/sg2_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylegan2/sg2_model.py -------------------------------------------------------------------------------- /model/lib/stylenerf/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/LICENSE.txt -------------------------------------------------------------------------------- /model/lib/stylenerf/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/README.md -------------------------------------------------------------------------------- /model/lib/stylenerf/dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/dnnlib/__init__.py -------------------------------------------------------------------------------- /model/lib/stylenerf/dnnlib/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/dnnlib/camera.py -------------------------------------------------------------------------------- /model/lib/stylenerf/dnnlib/filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/dnnlib/filters.py -------------------------------------------------------------------------------- /model/lib/stylenerf/dnnlib/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/dnnlib/geometry.py -------------------------------------------------------------------------------- /model/lib/stylenerf/dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/dnnlib/util.py -------------------------------------------------------------------------------- /model/lib/stylenerf/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/generate.py -------------------------------------------------------------------------------- /model/lib/stylenerf/launcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/launcher.py -------------------------------------------------------------------------------- /model/lib/stylenerf/legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/legacy.py -------------------------------------------------------------------------------- /model/lib/stylenerf/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/renderer.py -------------------------------------------------------------------------------- /model/lib/stylenerf/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/requirements.txt -------------------------------------------------------------------------------- /model/lib/stylenerf/run_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/run_train.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/__init__.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/custom_ops.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/distributed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/distributed_utils.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/misc.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/__init__.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/bias_act.cpp -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/bias_act.cu -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/bias_act.h -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/bias_act.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/conv2d_resample.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/filtered_lrelu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/filtered_lrelu.cpp -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/filtered_lrelu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/filtered_lrelu.cu -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/filtered_lrelu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/filtered_lrelu.h -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/filtered_lrelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/filtered_lrelu.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/filtered_lrelu_ns.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/filtered_lrelu_ns.cu -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/filtered_lrelu_rd.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/filtered_lrelu_rd.cu -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/filtered_lrelu_wr.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/filtered_lrelu_wr.cu -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/fma.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/hash_sample.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/hash_sample.cpp -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/hash_sample.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/hash_sample.cu -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/hash_sample.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/hash_sample.h -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/hash_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/hash_sample.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/nerf_utils.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/nerf_utils.cu -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/nerf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/nerf_utils.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/upfirdn2d.h -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/upfirdn2d.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/ops/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/ops/utils.h -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/persistence.py -------------------------------------------------------------------------------- /model/lib/stylenerf/torch_utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/torch_utils/training_stats.py -------------------------------------------------------------------------------- /model/lib/stylenerf/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/training/__init__.py -------------------------------------------------------------------------------- /model/lib/stylenerf/training/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/training/augment.py -------------------------------------------------------------------------------- /model/lib/stylenerf/training/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/training/data_utils.py -------------------------------------------------------------------------------- /model/lib/stylenerf/training/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/training/dataset.py -------------------------------------------------------------------------------- /model/lib/stylenerf/training/facial_recognition/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved -------------------------------------------------------------------------------- /model/lib/stylenerf/training/facial_recognition/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/training/facial_recognition/helpers.py -------------------------------------------------------------------------------- /model/lib/stylenerf/training/facial_recognition/model_irse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/training/facial_recognition/model_irse.py -------------------------------------------------------------------------------- /model/lib/stylenerf/training/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/training/loss.py -------------------------------------------------------------------------------- /model/lib/stylenerf/training/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/training/networks.py -------------------------------------------------------------------------------- /model/lib/stylenerf/training/stylenerf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/training/stylenerf.py -------------------------------------------------------------------------------- /model/lib/stylenerf/training/training_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/training/training_loop.py -------------------------------------------------------------------------------- /model/lib/stylenerf/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/visualizer.py -------------------------------------------------------------------------------- /model/lib/stylenerf/viz/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/viz/__init__.py -------------------------------------------------------------------------------- /model/lib/stylenerf/viz/camera_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/viz/camera_widget.py -------------------------------------------------------------------------------- /model/lib/stylenerf/viz/capture_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/viz/capture_widget.py -------------------------------------------------------------------------------- /model/lib/stylenerf/viz/equivariance_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/viz/equivariance_widget.py -------------------------------------------------------------------------------- /model/lib/stylenerf/viz/latent_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/viz/latent_widget.py -------------------------------------------------------------------------------- /model/lib/stylenerf/viz/layer_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/viz/layer_widget.py -------------------------------------------------------------------------------- /model/lib/stylenerf/viz/performance_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/viz/performance_widget.py -------------------------------------------------------------------------------- /model/lib/stylenerf/viz/pickle_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/viz/pickle_widget.py -------------------------------------------------------------------------------- /model/lib/stylenerf/viz/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/viz/renderer.py -------------------------------------------------------------------------------- /model/lib/stylenerf/viz/stylemix_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/viz/stylemix_widget.py -------------------------------------------------------------------------------- /model/lib/stylenerf/viz/trunc_noise_widget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/lib/stylenerf/viz/trunc_noise_widget.py -------------------------------------------------------------------------------- /model/mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/mapping.py -------------------------------------------------------------------------------- /model/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/model_utils.py -------------------------------------------------------------------------------- /model/prompt_gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/prompt_gan.py -------------------------------------------------------------------------------- /model/prompt_gan_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/prompt_gan_base.py -------------------------------------------------------------------------------- /model/prompt_gan_pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/prompt_gan_pose.py -------------------------------------------------------------------------------- /model/vanilla_gan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/model/vanilla_gan.py -------------------------------------------------------------------------------- /preprocess/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /preprocess/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/preprocess/empty.py -------------------------------------------------------------------------------- /preprocess/empty_small_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/preprocess/empty_small_eval.py -------------------------------------------------------------------------------- /preprocess/pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/preprocess/pose.py -------------------------------------------------------------------------------- /preprocess/to_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/preprocess/to_model.py -------------------------------------------------------------------------------- /raw_data/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /raw_data/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/raw_data/empty.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/trainer/trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /utils/config_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/utils/config_utils.py -------------------------------------------------------------------------------- /utils/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/utils/file_utils.py -------------------------------------------------------------------------------- /utils/program_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/utils/program_utils.py -------------------------------------------------------------------------------- /utils/training_arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/utils/training_arguments.py -------------------------------------------------------------------------------- /visualization/__init__.py: -------------------------------------------------------------------------------- 1 | # Created by Chen Henry Wu 2 | -------------------------------------------------------------------------------- /visualization/empty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/visualization/empty.py -------------------------------------------------------------------------------- /visualization/pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/visualization/pose.py -------------------------------------------------------------------------------- /visualization/single_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenWu98/generative-visual-prompt/HEAD/visualization/single_image.py --------------------------------------------------------------------------------