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
234 stars 24 forks source link

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

Open bleachyin opened 4 months ago

bleachyin commented 4 months ago

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

5632515 commented 1 day ago

输出 channel 和目标的二值化图片一样就好