├── .gitignore ├── README.md ├── common_flags.py ├── data ├── flying_animals_data │ └── fa_data_decode.py ├── flying_animals_utils.py ├── multi_dsprites_utils.py ├── multi_texture_data │ ├── bg.png │ ├── ellipse_2.png │ ├── square_2.png │ └── tex.png ├── multi_texture_utils.py └── objects_room_utils.py ├── doc ├── flying_animals.gif ├── multi_texture.gif ├── objects_room.gif └── pc.gif ├── eval └── eval_VAE.py ├── main.py ├── model ├── Summary.py ├── __init__.py ├── globalVAE_graph.py ├── nets.py ├── train_graph.py ├── traverse_graph.py └── utils │ ├── __init__.py │ ├── convolution_utils.py │ ├── generic_utils.py │ └── loss_utils.py ├── sample_imgs ├── flying_animals │ ├── 01.png │ └── 02.png ├── multi_dsprites │ ├── 01.png │ └── 02.png ├── multi_texture │ └── 01.png └── objects_room │ ├── 01.png │ ├── 02.png │ ├── 03.png │ └── 04.png ├── script ├── flying_animals │ ├── disentangle.sh │ ├── pretrain_inpainter.sh │ ├── test_segmentation.sh │ ├── train_CIS.sh │ └── train_VAE.sh ├── multi_dsprites │ ├── disentangle.sh │ ├── pretrain_inpainter.sh │ ├── test_segmentation.sh │ ├── train_CIS.sh │ └── train_VAE.sh ├── multi_texture │ ├── disentangle.sh │ ├── perceptual_consistency │ │ ├── finetune_PC.sh │ │ ├── test_segmentation.sh │ │ ├── train_CIS.sh │ │ └── train_VAE.sh │ ├── pretrain_inpainter.sh │ ├── test_segmentation.sh │ ├── train_CIS.sh │ └── train_VAE.sh └── objects_room │ ├── disentangle.sh │ ├── pretrain_inpainter.sh │ ├── test_segmentation.sh │ ├── train_CIS.sh │ └── train_VAE.sh ├── tb.sh ├── test_segmentation.py └── trainer ├── __init__.py ├── train_CIS.py ├── train_PC.py ├── train_VAE.py ├── train_end2end.py ├── train_globalVAE.py └── train_inpainter.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/README.md -------------------------------------------------------------------------------- /common_flags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/common_flags.py -------------------------------------------------------------------------------- /data/flying_animals_data/fa_data_decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/data/flying_animals_data/fa_data_decode.py -------------------------------------------------------------------------------- /data/flying_animals_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/data/flying_animals_utils.py -------------------------------------------------------------------------------- /data/multi_dsprites_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/data/multi_dsprites_utils.py -------------------------------------------------------------------------------- /data/multi_texture_data/bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/data/multi_texture_data/bg.png -------------------------------------------------------------------------------- /data/multi_texture_data/ellipse_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/data/multi_texture_data/ellipse_2.png -------------------------------------------------------------------------------- /data/multi_texture_data/square_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/data/multi_texture_data/square_2.png -------------------------------------------------------------------------------- /data/multi_texture_data/tex.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/data/multi_texture_data/tex.png -------------------------------------------------------------------------------- /data/multi_texture_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/data/multi_texture_utils.py -------------------------------------------------------------------------------- /data/objects_room_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/data/objects_room_utils.py -------------------------------------------------------------------------------- /doc/flying_animals.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/doc/flying_animals.gif -------------------------------------------------------------------------------- /doc/multi_texture.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/doc/multi_texture.gif -------------------------------------------------------------------------------- /doc/objects_room.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/doc/objects_room.gif -------------------------------------------------------------------------------- /doc/pc.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/doc/pc.gif -------------------------------------------------------------------------------- /eval/eval_VAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/eval/eval_VAE.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/main.py -------------------------------------------------------------------------------- /model/Summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/model/Summary.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | import sys 2 | sys.path.append('..') -------------------------------------------------------------------------------- /model/globalVAE_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/model/globalVAE_graph.py -------------------------------------------------------------------------------- /model/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/model/nets.py -------------------------------------------------------------------------------- /model/train_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/model/train_graph.py -------------------------------------------------------------------------------- /model/traverse_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/model/traverse_graph.py -------------------------------------------------------------------------------- /model/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/utils/convolution_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/model/utils/convolution_utils.py -------------------------------------------------------------------------------- /model/utils/generic_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/model/utils/generic_utils.py -------------------------------------------------------------------------------- /model/utils/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/model/utils/loss_utils.py -------------------------------------------------------------------------------- /sample_imgs/flying_animals/01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/sample_imgs/flying_animals/01.png -------------------------------------------------------------------------------- /sample_imgs/flying_animals/02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/sample_imgs/flying_animals/02.png -------------------------------------------------------------------------------- /sample_imgs/multi_dsprites/01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/sample_imgs/multi_dsprites/01.png -------------------------------------------------------------------------------- /sample_imgs/multi_dsprites/02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/sample_imgs/multi_dsprites/02.png -------------------------------------------------------------------------------- /sample_imgs/multi_texture/01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/sample_imgs/multi_texture/01.png -------------------------------------------------------------------------------- /sample_imgs/objects_room/01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/sample_imgs/objects_room/01.png -------------------------------------------------------------------------------- /sample_imgs/objects_room/02.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/sample_imgs/objects_room/02.png -------------------------------------------------------------------------------- /sample_imgs/objects_room/03.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/sample_imgs/objects_room/03.png -------------------------------------------------------------------------------- /sample_imgs/objects_room/04.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/sample_imgs/objects_room/04.png -------------------------------------------------------------------------------- /script/flying_animals/disentangle.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/flying_animals/disentangle.sh -------------------------------------------------------------------------------- /script/flying_animals/pretrain_inpainter.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/flying_animals/pretrain_inpainter.sh -------------------------------------------------------------------------------- /script/flying_animals/test_segmentation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/flying_animals/test_segmentation.sh -------------------------------------------------------------------------------- /script/flying_animals/train_CIS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/flying_animals/train_CIS.sh -------------------------------------------------------------------------------- /script/flying_animals/train_VAE.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/flying_animals/train_VAE.sh -------------------------------------------------------------------------------- /script/multi_dsprites/disentangle.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_dsprites/disentangle.sh -------------------------------------------------------------------------------- /script/multi_dsprites/pretrain_inpainter.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_dsprites/pretrain_inpainter.sh -------------------------------------------------------------------------------- /script/multi_dsprites/test_segmentation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_dsprites/test_segmentation.sh -------------------------------------------------------------------------------- /script/multi_dsprites/train_CIS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_dsprites/train_CIS.sh -------------------------------------------------------------------------------- /script/multi_dsprites/train_VAE.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_dsprites/train_VAE.sh -------------------------------------------------------------------------------- /script/multi_texture/disentangle.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_texture/disentangle.sh -------------------------------------------------------------------------------- /script/multi_texture/perceptual_consistency/finetune_PC.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_texture/perceptual_consistency/finetune_PC.sh -------------------------------------------------------------------------------- /script/multi_texture/perceptual_consistency/test_segmentation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_texture/perceptual_consistency/test_segmentation.sh -------------------------------------------------------------------------------- /script/multi_texture/perceptual_consistency/train_CIS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_texture/perceptual_consistency/train_CIS.sh -------------------------------------------------------------------------------- /script/multi_texture/perceptual_consistency/train_VAE.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_texture/perceptual_consistency/train_VAE.sh -------------------------------------------------------------------------------- /script/multi_texture/pretrain_inpainter.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_texture/pretrain_inpainter.sh -------------------------------------------------------------------------------- /script/multi_texture/test_segmentation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_texture/test_segmentation.sh -------------------------------------------------------------------------------- /script/multi_texture/train_CIS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_texture/train_CIS.sh -------------------------------------------------------------------------------- /script/multi_texture/train_VAE.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/multi_texture/train_VAE.sh -------------------------------------------------------------------------------- /script/objects_room/disentangle.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/objects_room/disentangle.sh -------------------------------------------------------------------------------- /script/objects_room/pretrain_inpainter.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/objects_room/pretrain_inpainter.sh -------------------------------------------------------------------------------- /script/objects_room/test_segmentation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/objects_room/test_segmentation.sh -------------------------------------------------------------------------------- /script/objects_room/train_CIS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/objects_room/train_CIS.sh -------------------------------------------------------------------------------- /script/objects_room/train_VAE.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/script/objects_room/train_VAE.sh -------------------------------------------------------------------------------- /tb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/tb.sh -------------------------------------------------------------------------------- /test_segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/test_segmentation.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trainer/train_CIS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/trainer/train_CIS.py -------------------------------------------------------------------------------- /trainer/train_PC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/trainer/train_PC.py -------------------------------------------------------------------------------- /trainer/train_VAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/trainer/train_VAE.py -------------------------------------------------------------------------------- /trainer/train_end2end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/trainer/train_end2end.py -------------------------------------------------------------------------------- /trainer/train_globalVAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/trainer/train_globalVAE.py -------------------------------------------------------------------------------- /trainer/train_inpainter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenYutongTHU/Learning-to-manipulate-individual-objects-in-an-image-Implementation/HEAD/trainer/train_inpainter.py --------------------------------------------------------------------------------