从带印记到干净:将带水印的图像转变为清晰的视觉效果
您是否想知道如何使用python从图像中去除水印?很简单!如果您有兴趣,您应该了解 python 并具备 cnn 和 tensorflow dl 框架等计算机视觉模型的基本知识,以便遵循架构!在运行代码之前,请确保您阅读了要去除水印的图像的版权法。
遵循的步骤 -
创建一个新google colab 笔记本。将运行时更改为t4 gpu,以增强计算能力来运行推理管道。
安装conda包,创建并激活conda环境
由于google colab使用最新的tensorflow和python版本,并且本项目使用python 3.6支持的tensorflow=1.15.0,因此在colab环境中安装miniconda
# set pythonpath%env pythonpath = # /env/python# set up miniconda and set the path '/usr/local'!wget https://repo.anaconda.com/miniconda/miniconda3-py38_4.12.0-linux-x86_64.sh!chmod +x miniconda3-py38_4.12.0-linux-x86_64.sh!./miniconda3-py38_4.12.0-linux-x86_64.sh -b -f -p /usr/localimport syssys.path.append('/usr/local/lib/python3.8/site-packages')# create a new conda environment using python 3.3!conda create -n myenv python=3.6
3.在env.中安装软件包
%%shelleval "$(conda shell.bash hook)"conda activate myenvconda install -y tensorflow==1.15 pillow opencv matplotlib pyyamlconda install -y tensorflow-gpupip install --upgrade pippip install git+https://github.com/jiahuiyu/neuralgym
4.克隆仓库
!git clone https://github.com/zuruoke/watermark-removal
5.从驱动器下载模型文件并将其粘贴到/watermark-removal/model目录中。
6.执行python代码从istock图像中删除水印。如果您有 alamy、shutterstock 或自定义水印图像,请在 utils// 中添加 mask.png。
%%shelleval "$(conda shell.bash hook)"conda activate myenvcd watermark-removalpython main.py --image path-to-input-image --output path-to-output-image --checkpoint_dir model/ --watermark_type istock