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