此回购包含AAAI 2021论文的代码和结果:
,
| | | (Google CoLab)
拟议的两阶段框架概述。 首先,我们提出了一个用于水印检测,删除和恢复的多任务网络SplitNet。 然后,我们提出了RefineNet,以使用预测的蒙版和从上一阶段恢复的背景对学习区域进行平滑处理。 因此,我们的网络可以在没有任何人工干预的情况下以端到端的方式进行培训。 注意,为清楚起见,我们没有显示所有编码器和解码器之间的任何跳过连接。
整个项目将于2021年1月(几乎)发布。
数据集
我们合成了四个不同的数据集进行训练和测试,您可以通过下载该数据集。
预训练模型
其他经过预先训练的模型仍在重组和上传中,它将很快发布。
演示版
可以在google colab中建立一个易于使用的在线演示。
本地演示将很快发布。
前提条件
pip install -r requirements.
1