├── .gitignore ├── LICENSE.txt ├── README.md ├── assets ├── edit_finegrained.png ├── editing1.png ├── inversion.PNG ├── pairwise.PNG ├── rosetta_neurons_ex.PNG └── teaser.png ├── calc_metrics.py ├── dnnlib ├── __init__.py └── util.py ├── feature_networks ├── clip │ ├── __init__.py │ ├── bpe_simple_vocab_16e6.txt.gz │ ├── clip.py │ ├── model.py │ └── simple_tokenizer.py ├── constants.py ├── pretrained_builder.py └── vit.py ├── legacy.py ├── mae.py ├── match.py ├── match_utils ├── dataset.py ├── helpers.py ├── layers.py ├── loading.py ├── matching.py ├── models.py ├── nethook.py ├── plotting.py ├── proggan.py ├── stats.py ├── universal.py ├── visualize_pairwisematch.py └── visualize_rosetta.py ├── notebooks ├── Visualize_Pairwise.ipynb ├── Visualize_Rosetta_Neurons.ipynb ├── inversion-stylegan2.ipynb ├── inversion-stylegan3.ipynb ├── match.ipynb ├── rosetta-inversion-stylegan-xl.ipynb ├── shift_zoom_copypaste.ipynb └── single_neuron_edit.ipynb ├── pg_modules ├── blocks.py ├── diffaug.py ├── discriminator.py ├── networks_fastgan.py ├── networks_stylegan2.py └── projector.py ├── requirements.txt ├── rosetta.yml ├── run_match.sh ├── stylegan2 ├── convert_weight.py ├── model.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 ├── styleganxl ├── diffaug.py ├── legacy.py └── metric_utils.py └── torch_utils ├── __init__.py ├── custom_ops.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 ├── upfirdn2d.cpp ├── upfirdn2d.cu ├── upfirdn2d.h └── upfirdn2d.py ├── persistence.py └── training_stats.py /.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/README.md -------------------------------------------------------------------------------- /assets/edit_finegrained.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/assets/edit_finegrained.png -------------------------------------------------------------------------------- /assets/editing1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/assets/editing1.png -------------------------------------------------------------------------------- /assets/inversion.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/assets/inversion.PNG -------------------------------------------------------------------------------- /assets/pairwise.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/assets/pairwise.PNG -------------------------------------------------------------------------------- /assets/rosetta_neurons_ex.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/assets/rosetta_neurons_ex.PNG -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /calc_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/calc_metrics.py -------------------------------------------------------------------------------- /dnnlib/__init__.py: -------------------------------------------------------------------------------- 1 | from .util import EasyDict, make_cache_dir_path 2 | -------------------------------------------------------------------------------- /dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/dnnlib/util.py -------------------------------------------------------------------------------- /feature_networks/clip/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /feature_networks/clip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/feature_networks/clip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /feature_networks/clip/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/feature_networks/clip/clip.py -------------------------------------------------------------------------------- /feature_networks/clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/feature_networks/clip/model.py -------------------------------------------------------------------------------- /feature_networks/clip/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/feature_networks/clip/simple_tokenizer.py -------------------------------------------------------------------------------- /feature_networks/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/feature_networks/constants.py -------------------------------------------------------------------------------- /feature_networks/pretrained_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/feature_networks/pretrained_builder.py -------------------------------------------------------------------------------- /feature_networks/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/feature_networks/vit.py -------------------------------------------------------------------------------- /legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/legacy.py -------------------------------------------------------------------------------- /mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/mae.py -------------------------------------------------------------------------------- /match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match.py -------------------------------------------------------------------------------- /match_utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/dataset.py -------------------------------------------------------------------------------- /match_utils/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/helpers.py -------------------------------------------------------------------------------- /match_utils/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/layers.py -------------------------------------------------------------------------------- /match_utils/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/loading.py -------------------------------------------------------------------------------- /match_utils/matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/matching.py -------------------------------------------------------------------------------- /match_utils/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/models.py -------------------------------------------------------------------------------- /match_utils/nethook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/nethook.py -------------------------------------------------------------------------------- /match_utils/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/plotting.py -------------------------------------------------------------------------------- /match_utils/proggan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/proggan.py -------------------------------------------------------------------------------- /match_utils/stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/stats.py -------------------------------------------------------------------------------- /match_utils/universal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/universal.py -------------------------------------------------------------------------------- /match_utils/visualize_pairwisematch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/visualize_pairwisematch.py -------------------------------------------------------------------------------- /match_utils/visualize_rosetta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/match_utils/visualize_rosetta.py -------------------------------------------------------------------------------- /notebooks/Visualize_Pairwise.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/notebooks/Visualize_Pairwise.ipynb -------------------------------------------------------------------------------- /notebooks/Visualize_Rosetta_Neurons.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/notebooks/Visualize_Rosetta_Neurons.ipynb -------------------------------------------------------------------------------- /notebooks/inversion-stylegan2.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/notebooks/inversion-stylegan2.ipynb -------------------------------------------------------------------------------- /notebooks/inversion-stylegan3.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/notebooks/inversion-stylegan3.ipynb -------------------------------------------------------------------------------- /notebooks/match.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/notebooks/match.ipynb -------------------------------------------------------------------------------- /notebooks/rosetta-inversion-stylegan-xl.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/notebooks/rosetta-inversion-stylegan-xl.ipynb -------------------------------------------------------------------------------- /notebooks/shift_zoom_copypaste.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/notebooks/shift_zoom_copypaste.ipynb -------------------------------------------------------------------------------- /notebooks/single_neuron_edit.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/notebooks/single_neuron_edit.ipynb -------------------------------------------------------------------------------- /pg_modules/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/pg_modules/blocks.py -------------------------------------------------------------------------------- /pg_modules/diffaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/pg_modules/diffaug.py -------------------------------------------------------------------------------- /pg_modules/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/pg_modules/discriminator.py -------------------------------------------------------------------------------- /pg_modules/networks_fastgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/pg_modules/networks_fastgan.py -------------------------------------------------------------------------------- /pg_modules/networks_stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/pg_modules/networks_stylegan2.py -------------------------------------------------------------------------------- /pg_modules/projector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/pg_modules/projector.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/requirements.txt -------------------------------------------------------------------------------- /rosetta.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/rosetta.yml -------------------------------------------------------------------------------- /run_match.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/run_match.sh -------------------------------------------------------------------------------- /stylegan2/convert_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/stylegan2/convert_weight.py -------------------------------------------------------------------------------- /stylegan2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/stylegan2/model.py -------------------------------------------------------------------------------- /stylegan2/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/stylegan2/op/__init__.py -------------------------------------------------------------------------------- /stylegan2/op/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/stylegan2/op/conv2d_gradfix.py -------------------------------------------------------------------------------- /stylegan2/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/stylegan2/op/fused_act.py -------------------------------------------------------------------------------- /stylegan2/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/stylegan2/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /stylegan2/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/stylegan2/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /stylegan2/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/stylegan2/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /stylegan2/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/stylegan2/op/upfirdn2d.py -------------------------------------------------------------------------------- /stylegan2/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/stylegan2/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /styleganxl/diffaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/styleganxl/diffaug.py -------------------------------------------------------------------------------- /styleganxl/legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/styleganxl/legacy.py -------------------------------------------------------------------------------- /styleganxl/metric_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/styleganxl/metric_utils.py -------------------------------------------------------------------------------- /torch_utils/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | # empty 4 | -------------------------------------------------------------------------------- /torch_utils/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/custom_ops.py -------------------------------------------------------------------------------- /torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/misc.py -------------------------------------------------------------------------------- /torch_utils/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/__init__.py -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/bias_act.cpp -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/bias_act.cu -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/bias_act.h -------------------------------------------------------------------------------- /torch_utils/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/bias_act.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/conv2d_resample.py -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/filtered_lrelu.cpp -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/filtered_lrelu.cu -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/filtered_lrelu.h -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/filtered_lrelu.py -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu_ns.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/filtered_lrelu_ns.cu -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu_rd.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/filtered_lrelu_rd.cu -------------------------------------------------------------------------------- /torch_utils/ops/filtered_lrelu_wr.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/filtered_lrelu_wr.cu -------------------------------------------------------------------------------- /torch_utils/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/fma.py -------------------------------------------------------------------------------- /torch_utils/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/upfirdn2d.h -------------------------------------------------------------------------------- /torch_utils/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/ops/upfirdn2d.py -------------------------------------------------------------------------------- /torch_utils/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/persistence.py -------------------------------------------------------------------------------- /torch_utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yossigandelsman/rosetta_neurons/HEAD/torch_utils/training_stats.py --------------------------------------------------------------------------------