psyker-team / mist-v2

A watermarking tool to protect artworks from AIGC-driven style mimicry (e.g. LoRA)
https://mist-project.github.io/
Apache License 2.0
447 stars 21 forks source link

不支持jpeg文件,只支持jpg #13

Closed AeroXi closed 10 months ago

AeroXi commented 10 months ago

现象: 如果输入图片是.jpeg后缀,会报错:

image

File "/home/mist/mist-v2/attacks/mist.py", line 978, in main perturbed_data, prompts, data_sizes = load_data( File "/home/mist/mist-v2/attacks/mist.py", line 542, in load_data images = torch.stack(images) RuntimeError: stack expects a non-empty TensorList 简单的解决办法是教程里让用户把jpeg重命名为jpg即可。 或许可以添加jpeg支持?毕竟都是同一种格式的不同后缀名