├── .gitignore ├── LICENSE ├── README.md ├── assets ├── full_architecture.png └── teaser.png ├── cogs ├── data │ ├── base_sketch_style_label.py │ └── sketch_style_label.py ├── models │ ├── cogs_transformer.py │ ├── vae.py │ └── vqgan.py ├── modules │ ├── autoencoder │ │ └── lpips │ │ │ └── vgg.pth │ ├── diffusionmodules │ │ └── model.py │ ├── discriminator │ │ └── model.py │ ├── losses │ │ ├── __init__.py │ │ ├── lpips.py │ │ ├── supcon_vae.py │ │ └── vqperceptual.py │ ├── style │ │ ├── aladin.py │ │ ├── hparams.py │ │ └── networks.py │ ├── transformer │ │ ├── mingpt.py │ │ └── permuter.py │ ├── util.py │ ├── vae │ │ └── model.py │ └── vqvae │ │ └── quantize.py └── util.py ├── configs ├── pseudosketches_cogs_transformer.yaml ├── pseudosketches_cogs_vae.yaml └── pseudosketches_vqgan.yaml ├── data └── pseudosketches │ ├── pseudosketches_train.txt │ └── pseudosketches_val.txt ├── environment.yaml ├── main.py ├── scripts └── sample_cogs_transformer.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/README.md -------------------------------------------------------------------------------- /assets/full_architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/assets/full_architecture.png -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /cogs/data/base_sketch_style_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/data/base_sketch_style_label.py -------------------------------------------------------------------------------- /cogs/data/sketch_style_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/data/sketch_style_label.py -------------------------------------------------------------------------------- /cogs/models/cogs_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/models/cogs_transformer.py -------------------------------------------------------------------------------- /cogs/models/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/models/vae.py -------------------------------------------------------------------------------- /cogs/models/vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/models/vqgan.py -------------------------------------------------------------------------------- /cogs/modules/autoencoder/lpips/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/autoencoder/lpips/vgg.pth -------------------------------------------------------------------------------- /cogs/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /cogs/modules/discriminator/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/discriminator/model.py -------------------------------------------------------------------------------- /cogs/modules/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/losses/__init__.py -------------------------------------------------------------------------------- /cogs/modules/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/losses/lpips.py -------------------------------------------------------------------------------- /cogs/modules/losses/supcon_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/losses/supcon_vae.py -------------------------------------------------------------------------------- /cogs/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /cogs/modules/style/aladin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/style/aladin.py -------------------------------------------------------------------------------- /cogs/modules/style/hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/style/hparams.py -------------------------------------------------------------------------------- /cogs/modules/style/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/style/networks.py -------------------------------------------------------------------------------- /cogs/modules/transformer/mingpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/transformer/mingpt.py -------------------------------------------------------------------------------- /cogs/modules/transformer/permuter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/transformer/permuter.py -------------------------------------------------------------------------------- /cogs/modules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/util.py -------------------------------------------------------------------------------- /cogs/modules/vae/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/vae/model.py -------------------------------------------------------------------------------- /cogs/modules/vqvae/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/modules/vqvae/quantize.py -------------------------------------------------------------------------------- /cogs/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/cogs/util.py -------------------------------------------------------------------------------- /configs/pseudosketches_cogs_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/configs/pseudosketches_cogs_transformer.yaml -------------------------------------------------------------------------------- /configs/pseudosketches_cogs_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/configs/pseudosketches_cogs_vae.yaml -------------------------------------------------------------------------------- /configs/pseudosketches_vqgan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/configs/pseudosketches_vqgan.yaml -------------------------------------------------------------------------------- /data/pseudosketches/pseudosketches_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/data/pseudosketches/pseudosketches_train.txt -------------------------------------------------------------------------------- /data/pseudosketches/pseudosketches_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/data/pseudosketches/pseudosketches_val.txt -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/environment.yaml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/main.py -------------------------------------------------------------------------------- /scripts/sample_cogs_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/scripts/sample_cogs_transformer.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cusuh/CoGS/HEAD/setup.py --------------------------------------------------------------------------------