├── CSD ├── __init__.py ├── loss_utils.py ├── losses.py ├── model.py ├── train_csd.py └── utils.py ├── LICENSE ├── README.md ├── __init__.py ├── artists_400.txt ├── data ├── laion.py └── wikiart.py ├── embeddings └── .gitkeep ├── environment.yaml ├── github_teaser.jpg ├── laion-styles-subset-tags.txt ├── main_sim.py ├── metrics ├── __init__.py └── metrics.py ├── models ├── clip │ ├── __init__.py │ ├── bpe_simple_vocab_16e6.txt.gz │ ├── clip.py │ ├── model.py │ └── simple_tokenizer.py ├── dino_vits.py └── moco_vits.py ├── pretrainedmodels └── .gitkeep ├── search.py ├── search ├── __init__.py ├── embeddings.py └── faiss_search.py ├── utils.py └── wikiart.csv /CSD/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /CSD/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/CSD/loss_utils.py -------------------------------------------------------------------------------- /CSD/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/CSD/losses.py -------------------------------------------------------------------------------- /CSD/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/CSD/model.py -------------------------------------------------------------------------------- /CSD/train_csd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/CSD/train_csd.py -------------------------------------------------------------------------------- /CSD/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/CSD/utils.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /artists_400.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/artists_400.txt -------------------------------------------------------------------------------- /data/laion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/data/laion.py -------------------------------------------------------------------------------- /data/wikiart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/data/wikiart.py -------------------------------------------------------------------------------- /embeddings/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/environment.yaml -------------------------------------------------------------------------------- /github_teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/github_teaser.jpg -------------------------------------------------------------------------------- /laion-styles-subset-tags.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/laion-styles-subset-tags.txt -------------------------------------------------------------------------------- /main_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/main_sim.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/metrics/metrics.py -------------------------------------------------------------------------------- /models/clip/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /models/clip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/models/clip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /models/clip/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/models/clip/clip.py -------------------------------------------------------------------------------- /models/clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/models/clip/model.py -------------------------------------------------------------------------------- /models/clip/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/models/clip/simple_tokenizer.py -------------------------------------------------------------------------------- /models/dino_vits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/models/dino_vits.py -------------------------------------------------------------------------------- /models/moco_vits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/models/moco_vits.py -------------------------------------------------------------------------------- /pretrainedmodels/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/search.py -------------------------------------------------------------------------------- /search/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /search/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/search/embeddings.py -------------------------------------------------------------------------------- /search/faiss_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/search/faiss_search.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/utils.py -------------------------------------------------------------------------------- /wikiart.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/learn2phoenix/CSD/HEAD/wikiart.csv --------------------------------------------------------------------------------