A self-supervised CNN for image watermark removal

Chunwei Tian, Menghua Zheng, Tiancai Jiao, Wangmeng Zuo, Yanning Zhang +1 more
3/9/2024
cs.CVeess.IV

No official code found. Best implementation: hellloxiaotian/SWCNN (74 stars, PyTorch, last commit 2024-08-21).

Abstract

Popular convolutional neural networks mainly use paired images in a supervised way for image watermark removal. However, watermarked images do not have reference images in the real world, which results in poor robustness of image watermark removal techniques. In this paper, we propose a self-supervised convolutional neural network (CNN) in image watermark removal (SWCNN). SWCNN uses a self-supervised way to construct reference watermarked images rather than given paired training samples, according to watermark distribution. A heterogeneous U-Net architecture is used to extract more complementary structural information via simple components for image watermark removal. Taking into account texture information, a mixed loss is exploited to improve visual effects of image watermark removal. Besides, a watermark dataset is conducted. Experimental results show that the proposed SWCNN is superior to popular CNNs in image watermark removal.

View on arXivView PDF

Code Implementations(1)

PyTorch

A self-supervised CNN for image watermark removal (IEEE Transactions on Circuits and Systems for Video 2024)

7416PythonMay 25, 20222 years ago
cnndeep-learningdeep-neural-networkspythonpytorch+3 more

Cite this paper

@article{tian2024selfsupervised,
  title  = {A self-supervised CNN for image watermark removal},
  author = {Chunwei Tian and Menghua Zheng and Tiancai Jiao and Wangmeng Zuo and Yanning Zhang and Chia-Wen Lin},
  year   = {2024},
  eprint = {2403.05807},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.05807v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.05807.svg)](https://codeofpaper.com/paper/2403.05807)

Discussion