硬件要求
lama-cleaner 支持 GPU 和 CPU 運(yùn)行乞旦，對(duì)于 GPU醇锚，需要約 2G 顯存哼御。
anaconda 的安裝和使用
見 http://www.reibang.com/p/adb1c163af46?v=1684159234141
創(chuàng)建虛擬環(huán)境 lama-cleaner
打開 anaconda navigator 軟件坯临，新建虛擬環(huán)境 lama-cleaner，進(jìn)入終端恋昼。
之后瀏覽器打開進(jìn)入 lama-cleaner 官網(wǎng) 看靠，按照官網(wǎng)提示安裝。
安裝 cuda 版本的 pytorch
官網(wǎng)提示：In order to use the GPU, install cuda version of pytorch first.
(lama-cleaner) conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
安裝依賴
(lama-cleaner) pip install lama-cleaner
啟動(dòng)服務(wù)
(lama-cleaner) lama-cleaner --model=lama --model-dir D:\project\lama-cleaner\models --output-dir D:\project\lama-cleaner\outputs --device=cuda --port=8888
啟動(dòng)過(guò)程中首先打印相關(guān)核心軟件版本液肌，之后挟炬，下載模型。
- Platform: Windows-10-10.0.22621-SP0
- Python version: 3.10.11
- torch: 2.0.1
- Pillow: 9.4.0
- diffusers: 0.14.0
- transformers: 4.27.4
- opencv-python: 4.7.0.72
- xformers: N/A
- accelerate: 0.19.0
- lama-cleaner: 1.1.2
Downloading: "https://github.com/Sanster/models/releases/download/add_big_lama/big-lama.pt" to D:\project\lama-cleaner\models\torch\hub\checkpoints\big-lama.pt
lama_cleaner.helper:download_model:52 - Download model success, md5: e3aa4aaa15225a33ec84f9f4bc47e500
lama_cleaner.helper:load_jit_model:102 - Loading model from: D:\project\lama-cleaner\models\torch\hub\checkpoints\big-lama.pt
Running on http://127.0.0.1:8888/
啟動(dòng)參數(shù)見：
https://lama-cleaner-docs.vercel.app/install/pip
使用服務(wù)
瀏覽器打開 http://127.0.0.1:8888/嗦哆，上傳圖片谤祖，涂抹水印，涂抹后等待水印去除即可老速。
image.png
