search

Royalvice / DocDiff

ACM Multimedia 2023: DocDiff: Document Enhancement via Residual Diffusion Models. Also contains 1597 red seals in Chinese scenes, along with their corresponding binary masks.
https://www.aibupt.com/
MIT License
196 stars 21 forks source link

如何训练一个二值化模型 #34

Open bleachyin opened 6 days ago

bleachyin commented 6 days ago

请问如何训练一个文档图片二值化模型，是采用config.yaml来配置样本集和训练集，这个docdiff模型可以直接进行end2end的训练对么？