├── .github └── ISSUE_TEMPLATE │ ├── bug_report.md │ ├── custom.md │ └── feature_request.md ├── .gitignore ├── LICENSE ├── README.md ├── data ├── __init__.py ├── ade20k_dataset.py ├── base_data_loader.py ├── base_dataset.py ├── cityscape_dataset.py ├── custom_dataset_data_loader.py ├── data_loader.py ├── image_folder.py └── segmentation_dataset.py ├── models ├── Discriminator_NET.py ├── MaskContextAE_NET.py ├── MaskTwoStreamConvSwitch_NET.py ├── MaskTwoStreamConv_NET.py ├── Pix2Pix_NET.py ├── TwoStreamAE_mask.py ├── __init__.py ├── base_model.py ├── joint_inference_model.py ├── layer_util.py ├── losses.py ├── mask_losses.py ├── models.py ├── networks.py ├── pix2pixHD_condImgColor_model.py ├── pix2pixHD_condImg_model.py └── sn_utils.py ├── options ├── __init__.py ├── box2mask_base_options.py ├── box2mask_test_options.py ├── box2mask_train_options.py ├── mask2image_base_options.py ├── mask2image_test_options.py └── mask2image_train_options.py ├── preprocess_ade.py ├── preprocess_city.py ├── scripts ├── download_pretrained_box2mask_ade.sh ├── download_pretrained_box2mask_city.sh ├── download_pretrained_mask2image_ade.sh ├── download_pretrained_mask2image_city.sh ├── test_joint_inference_city.sh ├── test_pretrained_box2mask_ade.sh ├── test_pretrained_box2mask_city.sh ├── test_pretrained_mask2image_ade.sh ├── test_pretrained_mask2image_city.sh ├── train_box2mask_ade.sh ├── train_box2mask_city.sh ├── train_mask2image_ade.sh └── train_mask2image_city.sh ├── setup.sh ├── train_box2mask.py ├── train_mask2image.py ├── util ├── __init__.py ├── data_util.py ├── html.py ├── image_pool.py ├── util.py └── visualizer.py ├── vis_box2mask.py ├── vis_joint_inference.py └── vis_mask2image.py /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/custom.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/.github/ISSUE_TEMPLATE/custom.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/ade20k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/data/ade20k_dataset.py -------------------------------------------------------------------------------- /data/base_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/data/base_data_loader.py -------------------------------------------------------------------------------- /data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/data/base_dataset.py -------------------------------------------------------------------------------- /data/cityscape_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/data/cityscape_dataset.py -------------------------------------------------------------------------------- /data/custom_dataset_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/data/custom_dataset_data_loader.py -------------------------------------------------------------------------------- /data/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/data/data_loader.py -------------------------------------------------------------------------------- /data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/data/image_folder.py -------------------------------------------------------------------------------- /data/segmentation_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/data/segmentation_dataset.py -------------------------------------------------------------------------------- /models/Discriminator_NET.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/Discriminator_NET.py -------------------------------------------------------------------------------- /models/MaskContextAE_NET.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/MaskContextAE_NET.py -------------------------------------------------------------------------------- /models/MaskTwoStreamConvSwitch_NET.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/MaskTwoStreamConvSwitch_NET.py -------------------------------------------------------------------------------- /models/MaskTwoStreamConv_NET.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/MaskTwoStreamConv_NET.py -------------------------------------------------------------------------------- /models/Pix2Pix_NET.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/Pix2Pix_NET.py -------------------------------------------------------------------------------- /models/TwoStreamAE_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/TwoStreamAE_mask.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/joint_inference_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/joint_inference_model.py -------------------------------------------------------------------------------- /models/layer_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/layer_util.py -------------------------------------------------------------------------------- /models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/losses.py -------------------------------------------------------------------------------- /models/mask_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/mask_losses.py -------------------------------------------------------------------------------- /models/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/models.py -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/networks.py -------------------------------------------------------------------------------- /models/pix2pixHD_condImgColor_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/pix2pixHD_condImgColor_model.py -------------------------------------------------------------------------------- /models/pix2pixHD_condImg_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/pix2pixHD_condImg_model.py -------------------------------------------------------------------------------- /models/sn_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/models/sn_utils.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /options/box2mask_base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/options/box2mask_base_options.py -------------------------------------------------------------------------------- /options/box2mask_test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/options/box2mask_test_options.py -------------------------------------------------------------------------------- /options/box2mask_train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/options/box2mask_train_options.py -------------------------------------------------------------------------------- /options/mask2image_base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/options/mask2image_base_options.py -------------------------------------------------------------------------------- /options/mask2image_test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/options/mask2image_test_options.py -------------------------------------------------------------------------------- /options/mask2image_train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/options/mask2image_train_options.py -------------------------------------------------------------------------------- /preprocess_ade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/preprocess_ade.py -------------------------------------------------------------------------------- /preprocess_city.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/preprocess_city.py -------------------------------------------------------------------------------- /scripts/download_pretrained_box2mask_ade.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/download_pretrained_box2mask_ade.sh -------------------------------------------------------------------------------- /scripts/download_pretrained_box2mask_city.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/download_pretrained_box2mask_city.sh -------------------------------------------------------------------------------- /scripts/download_pretrained_mask2image_ade.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/download_pretrained_mask2image_ade.sh -------------------------------------------------------------------------------- /scripts/download_pretrained_mask2image_city.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/download_pretrained_mask2image_city.sh -------------------------------------------------------------------------------- /scripts/test_joint_inference_city.sh: -------------------------------------------------------------------------------- 1 | python vis_joint_inference.py --gpu 0 2 | -------------------------------------------------------------------------------- /scripts/test_pretrained_box2mask_ade.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/test_pretrained_box2mask_ade.sh -------------------------------------------------------------------------------- /scripts/test_pretrained_box2mask_city.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/test_pretrained_box2mask_city.sh -------------------------------------------------------------------------------- /scripts/test_pretrained_mask2image_ade.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/test_pretrained_mask2image_ade.sh -------------------------------------------------------------------------------- /scripts/test_pretrained_mask2image_city.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/test_pretrained_mask2image_city.sh -------------------------------------------------------------------------------- /scripts/train_box2mask_ade.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/train_box2mask_ade.sh -------------------------------------------------------------------------------- /scripts/train_box2mask_city.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/train_box2mask_city.sh -------------------------------------------------------------------------------- /scripts/train_mask2image_ade.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/train_mask2image_ade.sh -------------------------------------------------------------------------------- /scripts/train_mask2image_city.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/scripts/train_mask2image_city.sh -------------------------------------------------------------------------------- /setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/setup.sh -------------------------------------------------------------------------------- /train_box2mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/train_box2mask.py -------------------------------------------------------------------------------- /train_mask2image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/train_mask2image.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /util/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/util/data_util.py -------------------------------------------------------------------------------- /util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/util/html.py -------------------------------------------------------------------------------- /util/image_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/util/image_pool.py -------------------------------------------------------------------------------- /util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/util/util.py -------------------------------------------------------------------------------- /util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/util/visualizer.py -------------------------------------------------------------------------------- /vis_box2mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/vis_box2mask.py -------------------------------------------------------------------------------- /vis_joint_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/vis_joint_inference.py -------------------------------------------------------------------------------- /vis_mask2image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xcyan/neurips18_hierchical_image_manipulation/HEAD/vis_mask2image.py --------------------------------------------------------------------------------