k8xu / ImageAttribution

Detecting Image Attribution for Text-to-Image Diffusion Models in RGB and Beyond
9 stars 0 forks source link

Demo model #1

Open harelber opened 3 months ago

harelber commented 3 months ago

Hi, Thanks for sharing the model. Is this model the image attributor or the model trained on the original RGB image?

k8xu commented 3 months ago

Thank you for your interest in our work! The demo model is for the 13-way classification task in Section 4.1, and the input is an original RGB image.