AID-AppEAL
收藏AID-AppEAL 数据集概述
数据集
AID-AppEAL 系统包含多个数据集,用于内容吸引力评估和增强。数据集存储在 datasets/ 目录下,用户需要从指定链接下载数据集文件(如 datasets/<dataset_name>.zip),并将其解压到数据集根目录中。
训练
相对内容吸引力评分比较器
训练相对内容吸引力评分比较器使用以下命令: bash bash scripts/1_relative_appeal_score_comparison.sh <dataset_name>
例如: bash bash scripts/1_relative_appeal_score_comparison.sh food bash scripts/1_relative_appeal_score_comparison.sh room
提供的检查点文件存储在 ckpts/pair_with_clip_<dataset_name>/last-v1.zip 路径下。
绝对内容吸引力评分预测器
训练绝对内容吸引力评分预测器使用以下命令: bash bash scripts/2_appeal_score_prediction.sh <dataset_name>
提供的检查点文件存储在 ckpts/singular_with_clip_<dataset_name>/last-v1.zip 路径下。
推理
内容吸引力评估和热图生成
仅评估内容吸引力评分: bash python appeal_heatmap_generation.py --name singular_with_clip_<dataset_name> --input_dir <path_to_input_images>
评估内容吸引力评分并生成热图: bash python appeal_heatmap_generation.py --name singular_with_clip_<dataset_name> --input_dir <path_to_input_images> --get_appeal_heatmap
结果默认保存在 outputs 目录下。
内容吸引力增强
使用 Automatic1111 的 Stable Diffusion web UI 进行内容吸引力增强,具体步骤包括使用热图作为修复掩码。
参考文献
bibtex @misc{chen2024aidappealautomaticimagedataset, title={AID-AppEAL: Automatic Image Dataset and Algorithm for Content Appeal Enhancement and Assessment Labeling}, author={Sherry X. Chen and Yaron Vaxman and Elad Ben Baruch and David Asulin and Aviad Moreshet and Misha Sra and Pradeep Sen}, year={2024}, eprint={2407.05546}, archivePrefix={arXiv}, primaryClass={cs.CV}, url={https://arxiv.org/abs/2407.05546}, }




