OpenXLab 国内入口 + Colab 免费笔记本:海外网络不好时的替代方案
OpenXLab 国内入口
OpenXLab 是商汤科技旗下的 AI 模型托管平台,CodeFormer 作者在上面也维护了官方部署:
适用场景:
- 从国内访问 HuggingFace / Replicate 速度慢或不稳定
- 更喜欢中文界面
- 不想注册海外账号
界面和参数与 HuggingFace Demo 基本一致,Gradio 框架统一的。直接打开、上传、调参数、出图。
OpenXLab 还提供免费 GPU 实例(需要登录并申请额度),可以在线跑完整 Python 脚本,不只是 Gradio Demo。
Google Colab 笔记本
CodeFormer 官方提供了 Colab 笔记本(README 里有链接),在免费的 T4 GPU 上可以跑完整推理流程。
优点
- 免费 GPU:Colab 免费版提供 T4(16GB 显存),足够跑 CodeFormer 整图处理
- 完整脚本:不是 Gradio 界面,是可以修改的 Python 代码
- 可上传本地图片:直接拖到 Colab 文件浏览器
- 国内可访问:Colab 在国内比 HuggingFace 好得多(有梯子的话)
基本使用流程
# 1. 克隆仓库
!git clone <仓库地址>
%cd CodeFormer
# 2. 安装依赖
!pip install basicsr facexlib realesrgan
!pip install -r requirements.txt
# 3. 下载预训练模型
!python scripts/download_pretrained_models.py facelib
!python scripts/download_pretrained_models.py CodeFormer
# 4. 上传图片到 inputs/TestWhole/ 目录
# Colab 左侧文件浏览器 → 右键 → 上传
# 5. 跑推理
!python inference_codeformer.py -w 0.7 \
--bg_upsampler realesrgan \
--input_path inputs/TestWhole/
# 6. 下载结果
# 结果在 results/ 目录,右键下载
Colab 的限制
- GPU 定时回收:免费版每次会话 12 小时,GPU 空闲 90 分钟会断开
- 显存有限:T4 16GB,整图 + 背景增强可能接近上限(见 显存不足排查)
- 视频处理慢:T4 处理视频是逐帧的,长视频建议分段
- ffmpeg 需要额外装:Colab 默认没有 ffmpeg,视频处理前要
!apt-get install ffmpeg
Colab Pro
如果经常用,考虑 Colab Pro($10/月):
- GPU 不会因为空闲被断开
- 更长的会话时间(24h)
- 优先分配更强的 GPU(有时能拿到 A100)
Colab 国内替代
从国内访问 Google Colab 通常需要梯子。如果不方便:
- AutoDL / 矩池云:国内 GPU 租赁平台,按小时计费(1~3 元/小时),预装 PyTorch + CUDA,只需
git clone CodeFormer即可 - 阿里云 / 腾讯云 GPU:按量付费,更灵活
- OpenXLab 在线实例:OpenXLab 也有 Jupyter 风格的在线实例,国内直连
在线 vs 本地,怎么选
| 维度 | 在线(Demo + API + Colab) | 本地自建 |
|---|---|---|
| 首次体验 | 秒级启动 | 30~60 分钟装环境 |
| 批量处理 | API 平台划算 | 自建 GPU 划算 |
| 视频处理 | Replicate 支持但贵 | 本地免费但慢 |
| 敏感数据 | 不推荐上传 | 完全本地化 |
| 离线使用 | 不行 | 可以 |
| 长期成本 | 按次付费 | GPU 硬件一次性投入 |
对于大多数个人用户,先在 HuggingFace / Colab 上验证,确实需要批量处理再考虑本地搭建。