WatermarkRemover-AI:一款開源的AI去水印工具
WatermarkRemover-ai是什么?
WatermarkRemover-AI是一款開源、實用的AI去水印工具。它利用微軟 Florence-2 技術檢測水印,并借助 LaMA 模型進行修復,能有效去除各類水印。支持單張圖片處理和批量操作,修復效果自然,還支持多種輸出格式。無需 GPU 也能順暢運行,還提供一鍵安裝腳本和簡單易用的界面。
WatermarkRemover-AI主要功能
雙模式操作:
支持處理單個圖像或整個目錄中的圖像。
先進的水印檢測:
使用 Florence-2 的開放詞匯檢測功能,能夠準確識別水印。
無縫修復:
使用 LaMA 模型進行上下文感知的修復,自然地填補去除水印后的區(qū)域。
自定義輸出:
可以配置水印檢測的最大邊界框大小。
可以設置水印區(qū)域的透明度。
可以強制輸出為特定格式(PNG、WEBP、JPG)。
進度跟蹤:
在 GUI 和 CLI 模式下均提供實時進度更新。
暗色模式支持:
GUI 會自動根據(jù)系統(tǒng)設置調(diào)整為暗色模式。
高效資源管理:
支持使用 CUDA 進行 GPU 加速(可選)。
WatermarkRemover-AI技術架構
Florence-2:用于水印檢測,通過開放詞匯對象檢測識別水印。
LaMA:用于修復,通過上下文感知的方式填補去除水印后的區(qū)域。
PyQt6:用于構建用戶友好的圖形界面。
WatermarkRemover-AI安裝方法
克隆倉庫:
git clone https://github.com/D-Ogi/WatermarkRemover-AI.git cd WatermarkRemover-AI
運行安裝腳本:
使用 setup.sh 腳本自動設置環(huán)境、安裝依賴并啟動 GUI 應用程序。
也可以通過命令行直接使用 CLI 模式。
手動激活環(huán)境:
如果需要手動激活環(huán)境,可以運行:
conda activate py312aiwatermark
WatermarkRemover-AI使用方法
圖形界面(GUI):
啟動 GUI 后,可以選擇處理單個圖像或整個目錄。
可以設置輸入/輸出路徑、配置選項(如透明度、邊界框大小等)。
點擊“開始”按鈕即可開始處理。
命令行界面(CLI):
基本命令格式:
python remwm.py input_path output_path [options]
常用選項:
--overwrite:覆蓋已存在的文件。
--transparent:將水印區(qū)域設置為透明。
--max-bbox-percent:設置水印檢測的最大邊界框大小(默認為 10%)。
--force-format:強制輸出為特定格式(PNG、WEBP 或 JPG)。
示例命令
python remwm.py ./input_images ./output_images --overwrite --transparent --max-bbox-percent=15 --force-format=PNG
github:https://github.com/D-Ogi/WatermarkRemover-AI