欢迎访问我的网站,希望内容对您有用,感兴趣的可以加入我们的社群。

开源图像上色项目IC-Light

算法 迷途小书童 7个月前 (06-04) 560次浏览 0个评论

前言

IC-LightImposing Consistent Light 的缩写,这个开源项目能够在不改变图片内容的情况下,控制生成图片的光照效果,也就是给图片上色。因此,这个项目可以用在给老照片上色。

IC-Light 发布了两个模型:文本条件(Text-Conditioned)模型和背景条件(Background-Conditioned)模型,这两种模型都将前景图像作为模型输入。

线上体验

我们可以登录站点 https://huggingface.co/spaces/lllyasviel/IC-Light 来体验模型的效果

iclight

在左侧栏上传一张图片,Prompt 栏填写上提示词,可以参考下方的一些常用提示词,这里就选个 beautiful woman, detailed faceLighting Preference 可以选择光照的位置,这里我没有特别需求,就默认 None,最后点击 Relight 开始

huggingface 默认使用 gpu 来处理,整理速度还是很快的,最后的效果如下

iclight

IC-Light 可调的参数还是蛮多的,大家可以根据自己的实际要求去调整,找到最佳的方案。

本地部署

如果你想进行离线部署的话,可以按照下面的命令步骤进行,我这里使用 anaconda 的环境,关于它的安装,请参考 Anaconda基本使用

# 拉取代码
git clone https://github.com/lllyasviel/IC-Light.git

# 进入源码目录
cd IC-Light

# 创建全新的python虚拟环境
conda create -n iclight python=3.10

# 激活上面创建的虚拟环境
conda activate iclight

# 安装gpu版本的torch和torchvision
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121

# 安装其它依赖库
pip install -r requirements.txt

# 运行demo程序
python gradio_demo.py

脚本运行过程中,会自动取下载模型

iclight

这个时候,就可以在浏览器中输入地址 http://127.0.0.1:6780

iclight

我在部署的过程中碰到了如下的错误,启动没问题,但是在处理的时候就出现了莫名的 error,后面发现是科学上网引起的,解决的办法就是,在执行 gradio_demo.py 脚本前,将科学上网关闭

iclight

这时候就可以正常处理了,命令行中也有进度显示。

iclight

软件包下载

提供一份可用的软件包下载,包括了需另外下载的模型文件,https://pan.quark.cn/s/b1db0efa1cee

参考资料

喜欢 (0)

您必须 登录 才能发表评论!