https://github.com/hellloxiaotian/swcnn

A self-supervised CNN for image watermark removal (IEEE Transactions on Circuits and Systems for Video 2024)
https://github.com/hellloxiaotian/swcnn

cnn deep-learning deep-neural-networks python pytorch self-supervised-learning unet watermark-removal

Last synced: 2 months ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/hellloxiaotian/swcnn
• Owner: hellloxiaotian
• Created: 2022-05-25T06:32:44.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2024-08-21T15:22:48.000Z (5 months ago)
• Last Synced: 2024-08-22T17:41:25.078Z (5 months ago)
• Topics: cnn, deep-learning, deep-neural-networks, python, pytorch, self-supervised-learning, unet, watermark-removal
• Language: Python
• Homepage:
• Size: 46.6 MB
• Stars: 31
• Watchers: 2
• Forks: 6
• Open Issues: 2
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
## This paper as well as  A self-supervised CNN for image watermark removal is conducted by Chunwei Tian, Menghua Zheng, Tiancai Jiao, Wangmeng Zuo, Yanning Zhang, Chia-Wen Lin. This  paper is accepted by the IEEE Transactions on Circuits and Systems for Video Technology (IF:8.4) and it is implemented by Pytorch. 

## Its abstract is shown as follows.

## 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.

## Requirements (Pytorch)

* pytorch==1.11.0

* tensorboard==2.9.1

* scikit-image==0.19.3

* pyyaml==5.1

* h5py==3.7.0

* opencv-python==4.6.0

* matplotlib==3.5.2

## DataSet

### Training datasets

#### The training dataset of the PASCAL VOC 2021 can be obtained at https://pjreddie.com/projects/pascal-voc-dataset-mirror/

### Training datasets

#### The training dataset of the PASCAL VOC 2012 can be obtained at 链接：https://pan.baidu.com/s/1At7c2egxalV1T8HpBGGQdw (56r7) or https://drive.google.com/drive/folders/1Byi-wwGwks69fItGesEHDBdgUr1xGFc5?usp=drive_link (Google drive).

## Command

Download model file at 链接：https://pan.baidu.com/s/1P-Z2fHIwUXIqghpH5gykkg (passwd: iu5i) and put `data`/  into `SWCNN/` or https://drive.google.com/drive/folders/1Byi-wwGwks69fItGesEHDBdgUr1xGFc5?usp=drive_link (Google drive).

### Test

python test.py --alpha 0.3  --loss L1 --self_supervised True  --PN True  --display False

### Train

python train.py --net HN --loss L1 --self_supervised True --PN True --alpha 0.3

## 1. Network architecture of SWCNN.

![image-20240304100316953](assets/image-20240304100316953.png)

## 2. Removing watermark performance of different methods for transparency of 0.3. 

![image-20240304100351018](assets/image-20240304100351018.png)

## 3. PSNR(dB) of different methods for different transparency. 

![image-20240304100545625](assets/image-20240304100545625.png)

## 4. NIQE and ILNIQE of different methods for transparency of 0.3.

![image-20240304101626408](assets/image-20240304101626408.png)

## 5. Visual images of different methods

![image-20240304101834486](assets/image-20240304101834486.png)

### 6. Visual images of different methods.

![image-20240304101912376](assets/image-20240304101912376.png)

### 7. Visual images of different methods.



### 8. Visual images of different methods.

![image-20240304102041706](assets/image-20240304102041706.png)

### 9. Visual images of different methods.

![image-20240304102055963](assets/image-20240304102055963.png)

## You can cite this paper by the following ways.

### 1. Chunwei Tian, Menghua Zheng, Tiancai Jiao, Wangmeng Zuo, Yanning Zhang, Chia-Wen Lin. A self-supervised CNN for image watermark removal[J]. IEEE Transactions on Circuits and Systems for Video Technology, 2024.

### 2.@article{tian2024A,

       title={A self-supervised CNN for image watermark removal},

       author={Tian, Chunwei and Zheng, Menghua and Jiao, Tiancai and Zuo, Wangmeng and Zhang, Yanning and Lin, Chia-Wen},

       journal={IEEE Transactions on Circuits and Systems for Video Technology},

      year={2024},

     publisher={IEEE}

}