├── LICENSE ├── README.md ├── asset ├── data │ ├── base.py │ ├── coco.py │ └── landscape.py ├── lr_scheduler.py ├── models │ ├── cond_transformer.py │ └── vqgan.py ├── modules │ ├── autoencoder │ │ └── lpips │ │ │ └── vgg.pth │ ├── diffusionmodules │ │ └── model.py │ ├── discriminator │ │ └── model.py │ ├── losses │ │ ├── __init__.py │ │ ├── lpips.py │ │ ├── segmentation.py │ │ └── vqperceptual.py │ ├── transformer │ │ ├── asset.py │ │ └── permuter.py │ ├── util.py │ └── vqvae │ │ └── quantize.py └── util.py ├── configs ├── coco_SGA_transformer_512.yaml ├── coco_guiding_transformer.yaml ├── coco_test.yaml ├── coco_vqgan.yaml ├── landscape_SGA_transformer_1024.yaml ├── landscape_SGA_transformer_512.yaml ├── landscape_guiding_transformer.yaml ├── landscape_test.yaml └── landscape_vqgan.yaml ├── data └── landscape │ ├── imgs │ ├── test │ │ └── landscape_input.jpg │ └── train │ │ └── landscape_input.jpg │ ├── seg │ ├── test_seg_256 │ │ └── landscape_input.png │ ├── test_seg_512 │ │ └── landscape_input.png │ ├── train_seg_256 │ │ └── landscape_input.png │ └── train_seg_512 │ │ └── landscape_input.png │ ├── test.txt │ └── train.txt ├── data_test ├── coco_input.png ├── coco_mask.png ├── coco_seg.png ├── landscape_input.jpg ├── landscape_mask.png └── landscape_seg.png ├── main.py ├── main_utils.py ├── pl_utils.py ├── results └── .gitignore └── test.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/README.md -------------------------------------------------------------------------------- /asset/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/data/base.py -------------------------------------------------------------------------------- /asset/data/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/data/coco.py -------------------------------------------------------------------------------- /asset/data/landscape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/data/landscape.py -------------------------------------------------------------------------------- /asset/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/lr_scheduler.py -------------------------------------------------------------------------------- /asset/models/cond_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/models/cond_transformer.py -------------------------------------------------------------------------------- /asset/models/vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/models/vqgan.py -------------------------------------------------------------------------------- /asset/modules/autoencoder/lpips/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/modules/autoencoder/lpips/vgg.pth -------------------------------------------------------------------------------- /asset/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /asset/modules/discriminator/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/modules/discriminator/model.py -------------------------------------------------------------------------------- /asset/modules/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/modules/losses/__init__.py -------------------------------------------------------------------------------- /asset/modules/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/modules/losses/lpips.py -------------------------------------------------------------------------------- /asset/modules/losses/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/modules/losses/segmentation.py -------------------------------------------------------------------------------- /asset/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /asset/modules/transformer/asset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/modules/transformer/asset.py -------------------------------------------------------------------------------- /asset/modules/transformer/permuter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/modules/transformer/permuter.py -------------------------------------------------------------------------------- /asset/modules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/modules/util.py -------------------------------------------------------------------------------- /asset/modules/vqvae/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/modules/vqvae/quantize.py -------------------------------------------------------------------------------- /asset/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/asset/util.py -------------------------------------------------------------------------------- /configs/coco_SGA_transformer_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/configs/coco_SGA_transformer_512.yaml -------------------------------------------------------------------------------- /configs/coco_guiding_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/configs/coco_guiding_transformer.yaml -------------------------------------------------------------------------------- /configs/coco_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/configs/coco_test.yaml -------------------------------------------------------------------------------- /configs/coco_vqgan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/configs/coco_vqgan.yaml -------------------------------------------------------------------------------- /configs/landscape_SGA_transformer_1024.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/configs/landscape_SGA_transformer_1024.yaml -------------------------------------------------------------------------------- /configs/landscape_SGA_transformer_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/configs/landscape_SGA_transformer_512.yaml -------------------------------------------------------------------------------- /configs/landscape_guiding_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/configs/landscape_guiding_transformer.yaml -------------------------------------------------------------------------------- /configs/landscape_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/configs/landscape_test.yaml -------------------------------------------------------------------------------- /configs/landscape_vqgan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/configs/landscape_vqgan.yaml -------------------------------------------------------------------------------- /data/landscape/imgs/test/landscape_input.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data/landscape/imgs/test/landscape_input.jpg -------------------------------------------------------------------------------- /data/landscape/imgs/train/landscape_input.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data/landscape/imgs/train/landscape_input.jpg -------------------------------------------------------------------------------- /data/landscape/seg/test_seg_256/landscape_input.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data/landscape/seg/test_seg_256/landscape_input.png -------------------------------------------------------------------------------- /data/landscape/seg/test_seg_512/landscape_input.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data/landscape/seg/test_seg_512/landscape_input.png -------------------------------------------------------------------------------- /data/landscape/seg/train_seg_256/landscape_input.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data/landscape/seg/train_seg_256/landscape_input.png -------------------------------------------------------------------------------- /data/landscape/seg/train_seg_512/landscape_input.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data/landscape/seg/train_seg_512/landscape_input.png -------------------------------------------------------------------------------- /data/landscape/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data/landscape/test.txt -------------------------------------------------------------------------------- /data/landscape/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data/landscape/train.txt -------------------------------------------------------------------------------- /data_test/coco_input.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data_test/coco_input.png -------------------------------------------------------------------------------- /data_test/coco_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data_test/coco_mask.png -------------------------------------------------------------------------------- /data_test/coco_seg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data_test/coco_seg.png -------------------------------------------------------------------------------- /data_test/landscape_input.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data_test/landscape_input.jpg -------------------------------------------------------------------------------- /data_test/landscape_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data_test/landscape_mask.png -------------------------------------------------------------------------------- /data_test/landscape_seg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/data_test/landscape_seg.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/main.py -------------------------------------------------------------------------------- /main_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/main_utils.py -------------------------------------------------------------------------------- /pl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/pl_utils.py -------------------------------------------------------------------------------- /results/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/results/.gitignore -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DifanLiu/ASSET/HEAD/test.py --------------------------------------------------------------------------------