lama cleaner

这里写自定义目录标题

  • 安装
  • 参数
    • 包含的额外plugins

安装

conda create --name lamacleaner python==3.10
pip install -r requirements.txt  
pip install gfpgan
pip install realesrgan
pip install rembg
pip install .

如果安装本package报错,可以尝试改:

packages=setuptools.find_packages("./") 

换成:

packages=setuptools.find_packages(".")

如果是GPU的,一定要装GPU版本的pytorch: (我用的11.8版本)

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

参数

默认的参数地址在:

lama_cleaner\parse_args.py

包含的额外plugins

这个工具包里包含着以下内容:

  • Interactive Segmentation
  • Background Removal
  • Anime Segmentation
  • RealESRGAN
  • GFPGAN
  • RestoreFormer
  • GIF

如果要使用以上的内容,需要在parse_args.py 这个文件里添加default=True
例如:

parser.add_argument(
        "--enable-realesrgan",
        default=True,
        action="store_true",
        help=REALESRGAN_HELP,
    )

把这些内容打开后,这个包会自动下载所有相关权重

lama cleaner_第1张图片

你可能感兴趣的:(计算机视觉)