├── AK4Prompts.py ├── AK4Prompts_pipeline.py ├── README.md ├── __pycache__ ├── AK4Prompts.cpython-38.pyc └── AK4Prompts_pipeline.cpython-38.pyc ├── aesthetic_scorer.py ├── config └── train_config.py ├── demo.ipynb ├── evaluation.py ├── images └── hps.png ├── keywords ├── __init__.py ├── data │ ├── keywords_collect_all.txt │ ├── keywords_list.txt │ └── negative.txt └── keywords_table.py ├── model ├── aesthetic │ └── sac+logos+ava1-l14-linearMSE.pth └── model_download.py ├── prompts ├── data.json ├── test_prompts_2000.json └── train_prompts.json ├── requirement.txt ├── test_prompt └── test_dataset_2000.json └── train.py /AK4Prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/AK4Prompts.py -------------------------------------------------------------------------------- /AK4Prompts_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/AK4Prompts_pipeline.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/README.md -------------------------------------------------------------------------------- /__pycache__/AK4Prompts.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/__pycache__/AK4Prompts.cpython-38.pyc -------------------------------------------------------------------------------- /__pycache__/AK4Prompts_pipeline.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/__pycache__/AK4Prompts_pipeline.cpython-38.pyc -------------------------------------------------------------------------------- /aesthetic_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/aesthetic_scorer.py -------------------------------------------------------------------------------- /config/train_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/config/train_config.py -------------------------------------------------------------------------------- /demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/demo.ipynb -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/evaluation.py -------------------------------------------------------------------------------- /images/hps.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/images/hps.png -------------------------------------------------------------------------------- /keywords/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /keywords/data/keywords_collect_all.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/keywords/data/keywords_collect_all.txt -------------------------------------------------------------------------------- /keywords/data/keywords_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/keywords/data/keywords_list.txt -------------------------------------------------------------------------------- /keywords/data/negative.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/keywords/data/negative.txt -------------------------------------------------------------------------------- /keywords/keywords_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/keywords/keywords_table.py -------------------------------------------------------------------------------- /model/aesthetic/sac+logos+ava1-l14-linearMSE.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/model/aesthetic/sac+logos+ava1-l14-linearMSE.pth -------------------------------------------------------------------------------- /model/model_download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/model/model_download.py -------------------------------------------------------------------------------- /prompts/data.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/prompts/data.json -------------------------------------------------------------------------------- /prompts/test_prompts_2000.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/prompts/test_prompts_2000.json -------------------------------------------------------------------------------- /prompts/train_prompts.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/prompts/train_prompts.json -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/requirement.txt -------------------------------------------------------------------------------- /test_prompt/test_dataset_2000.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/test_prompt/test_dataset_2000.json -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/woshidandan/AK4Prompts/HEAD/train.py --------------------------------------------------------------------------------