Image inpainting tool powered by SOTA AI Model. Remove any unwanted object, defect, people from your pictures or erase and replace(powered by stable diffusion) any thing on your pictures.
Go to file
2021-11-15 23:37:26 +01:00
lama_cleaner Added Dockerfile 2021-11-15 20:11:46 +01:00
.gitignore init 2021-11-15 22:21:01 +08:00
Dockerfile Added Dockerfile 2021-11-15 20:11:46 +01:00
LICENSE init 2021-11-15 22:21:01 +08:00
main.py Set TORCH_HOME 2021-11-15 23:37:26 +01:00
README.md Updated docker run gpu 2021-11-15 23:24:07 +01:00
requirements.txt init 2021-11-15 22:21:01 +08:00
setup.py init 2021-11-15 22:21:01 +08:00

Lama-cleaner: Image inpainting tool powered by LaMa

This project is mainly used for selfhosting LaMa model, some interaction improvements may be added later.

Quick Start

  • Install requirements: pip3 install -r requirements.txt
  • Start server: python3 main.py --device=cuda --port=8080

Development

Fronted

Frontend code are modified from cleanup.pictures, You can experience their great online services here.

  • Install dependencies:cd lama_cleaner/app/ && yarn
  • Start development server: yarn dev
  • Build: yarn build

Docker (cpu)

docker build -f Dockerfile -t lamacleaner .
docker run -p 8080:8080 -e cache_dir=/app/models -v models:/app/models -v $(pwd):/app --rm lamacleaner python3 main.py --device=cpu --port=8080

Docker (gpu)

docker build -f Dockerfile -t lamacleaner .
docker run --gpus all -p 8080:8080 -e cache_dir=/app/models -v models:/app/models -v $(pwd):/app --rm lamacleaner python3 main.py --device=cuda --port=8080

Then open http://localhost:8080