comfyui-Image-reward

comfyui-Image-reward
★ 33

人类偏好学习图像评分文本到图像排序与筛选
在ComfyUI中使用ImageReward模型，对生成图像按人类偏好进行评分，便于排序、筛选或作为训练奖励，提升生成结果的偏好一致性。
💡 为生成图像打分并按人类偏好排序或作为训练奖励
🍴 5 Forks💻 Python🔄 2024-06-14
📦
网盘下载
复制链接后前往夸克网盘下载
https://pan.quark.cn/s/51137d50651f
📦 requirements.txt
openai-clip
timm==0.6.13
fairscale==0.4.13
1711282071529
1711282130123
image
📄 README

comfyui-Image-reward

workflow example

Collaborate with mixlab-nodes to convert the workflow into an app.

配合mixlab-nodes，把workflow转为app使用。

ImageReward

Human preference learning in text-to-image generation. This is a paper for NeurIPS 2023, trained using the professional large-scale dataset ImageRewardDB: approximately 137,000 comparison pairs.

文本到图像生成中的人类偏好学习。这是NeurIPS 2023的一篇论文，训练使用专业的大规模数据集ImageRewardDB：约13.7万个⽐较pairs。

install

requirement.txt

models

ImageReward 放到

ComfyUI/models/image_reward/ImageReward.pt

ComfyUI/models/image_reward/med_config.json

bert-base-uncased 放到 ComfyUI/models/image_reward/bert-base-uncased/