https://github.com/hi-tech-ai/watermark-remove

Watermark Removal using TensorFlow
https://github.com/hi-tech-ai/watermark-remove

artefact-removal deep-learning skip-connections tensorflow watermark watermark-detection watermark-removal

Last synced: 15 days ago
JSON representation

Watermark Removal using TensorFlow

• Host: GitHub
• URL: https://github.com/hi-tech-ai/watermark-remove
• Owner: hi-tech-AI
• License: mit
• Created: 2024-04-12T19:48:20.000Z (about 2 years ago)
• Default Branch: tensorflow
• Last Pushed: 2025-02-14T16:14:55.000Z (over 1 year ago)
• Last Synced: 2025-02-25T02:40:30.735Z (over 1 year ago)
• Topics: artefact-removal, deep-learning, skip-connections, tensorflow, watermark, watermark-detection, watermark-removal
• Language: Python
• Homepage:
• Size: 1.55 MB
• Stars: 1
• Watchers: 1
• Forks: 2
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# Watermark-Removal

![version](https://img.shields.io/badge/version-v1.0.0-green.svg?style=plastic)

![pytorch](https://img.shields.io/badge/tensorflow-v1.15.0-green.svg?style=plastic)

![license](https://img.shields.io/badge/license-CC_BY--NC-green.svg?style=plastic)

An open source project that uses a machine learning based image inpainting methodology to remove watermark from images which is totally indistinguishable from the ground truth version of the image.

This project was inspired by the [Contextual Attention](https://arxiv.org/abs/1801.07892) (CVPR 2018) and [Gated Convolution](https://arxiv.org/abs/1806.03589) (ICCV 2019 Oral).

         

## Run

- use [Google colab](https://research.google.com/colaboratory/)

- First of all, clone this repo

      git clone https://github.com/morning120429/watermark-remove

- Change Directory to the repo

      cd watermark-removal

- Change Python version to 3.7 because tensorflow 1.15.0 does not support Python 3.8 or higher

- Since Google Colab uses the latest Tensorflow 2x version and this project uses 1.15.0, downgrade to Tensorflow 1.15.0 version and restart the runtime, (`although the new version of Google Colab does not need you to restart the runtime`).

      pip install tensorflow==1.15.0

- Install OpenCV 4.9.0.80

      pip install opencv-python==4.9.0.80

- Install tensorflow toolkit [neuralgym](https://github.com/JiahuiYu/neuralgym).

      pip install git+https://github.com/JiahuiYu/neuralgym

- Download the model dirs using this [link](https://drive.google.com/drive/folders/1xRV4EdjJuAfsX9pQme6XeoFznKXG0ptJ?usp=sharing) and put it under `model/` (rename `checkpoint.txt` to `checkpoint` because sometimes google drive automatically adds .txt after download)

And you're all Set!!

- Now remove the watermark on the image by runing the `main.py` file

      python main.py --image path-to-input-image --output path-to-output-image --checkpoint_dir model/ --watermark_type istock

## Citing

```

@article{yu2018generative,

  title={Generative Image Inpainting with Contextual Attention},

  author={Yu, Jiahui and Lin, Zhe and Yang, Jimei and Shen, Xiaohui and Lu, Xin and Huang, Thomas S},

  journal={arXiv preprint arXiv:1801.07892},

  year={2018}

}

@article{yu2018free,

  title={Free-Form Image Inpainting with Gated Convolution},

  author={Yu, Jiahui and Lin, Zhe and Yang, Jimei and Shen, Xiaohui and Lu, Xin and Huang, Thomas S},

  journal={arXiv preprint arXiv:1806.03589},

  year={2018}

}

```

## @tech_hosting

## License

This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.