此存储库提供 PromptFix 的官方 PyTorch 实现，包括预先训练的权重、训练和推理代码，以及用于训练的精选数据集。
📢 PromptFix 旨在按照人类指令处理降级的图像并删除不需要的元素。它支持广泛的任务，例如：
- 🎨 着色
- 🧹 对象删除
- 🌫️ 去雾
- 💨 去模糊
- 🖼️ 水印去除
- ❄️ 除雪
- 🌙 低光增强
PromptFix 建立在扩散模型骨架之上，利用 20 步去噪过程，在保留原始结构的同时，在纠正图像缺陷方面提供出色的性能。它还可以有效地泛化不同的纵横比。
我们策划了一个超过 100 万个样本的训练数据集。每个示例都包括配对图像以及说明和辅助文本提示。该数据集涵盖了多个低级图像处理任务。
注意：数据集打包成 Parquet 文件，由 100 个部分组成。每个零件都可以独立加载。如果您想在不下载整个数据集的情况下试验少量数据，则可以只下载几个 Parquet 文件。