RVM on Replicate

将 PeterL1n/RobustVideoMatting 封装为 Replicate 自定义模型，输入 mp4 视频，输出黑白 alpha-mask.mp4。

模型行为

• 输入：mp4 视频 URL 或本地文件
• 输出：黑白 alpha-mask.mp4
• 输出分辨率：与输入一致
• 输出帧率：与输入一致
• 输出音轨：不保留
• 模型变体：mobilenetv3

本地准备

~/.local/bin/cog login --token-stdin <<< "$REPLICATE_CLI_TOKEN"

本地调试

~/.local/bin/cog predict \
  -i input_video=https://test-videos.co.uk/vids/bigbuckbunny/mp4/h264/720/Big_Buck_Bunny_720_10s_1MB.mp4

推送

~/.local/bin/cog push r8.im/hcolde/rvm

线上调用

当前已验证可用的版本 ID：

a7698b75b09ca4ddf59f566fbd39da452c688654bc92391d802b6cbe6bd5bfc8

显式指定 version 的调用方式：

curl -s -X POST https://api.replicate.com/v1/predictions \
  -H "Authorization: Bearer $REPLICATE_API_TOKEN" \
  -H "Content-Type: application/json" \
  -d '{
    "version": "a7698b75b09ca4ddf59f566fbd39da452c688654bc92391d802b6cbe6bd5bfc8",
    "input": {
      "input_video": "https://test-videos.co.uk/vids/bigbuckbunny/mp4/h264/720/Big_Buck_Bunny_720_10s_1MB.mp4"
    }
  }'

查询最新版本 ID：

curl -s -H "Authorization: Bearer $REPLICATE_API_TOKEN" \
  https://api.replicate.com/v1/models/hcolde/rvm/versions | jq -r '.results[0].id'

已验证结果

• 推送模型：https://replicate.com/hcolde/rvm
• 已验证 prediction：https://replicate.com/p/gb8dcmkqahrgc0cx19evk2c89g
• 输出文件：https://replicate.delivery/czjl/8ba3LoXTvTqCG9GSinGPtIKwaL7J3JiYyRC4g7W2g3123qkF/alpha-mask.mp4
• 线上实测耗时：predict_time ≈ 20.99s，total_time ≈ 92.58s

创建模型

如果 hcolde/rvm 尚不存在，可先通过 Replicate API 创建：

curl -s -X POST https://api.replicate.com/v1/models \
  -H "Authorization: Bearer $REPLICATE_API_TOKEN" \
  -H "Content-Type: application/json" \
  -d '{
    "owner": "hcolde",
    "name": "rvm",
    "visibility": "public",
    "hardware": "gpu-t4"
  }'

说明

• 这次部署已经证明“显式 version 调用”可正常返回结果。
• 当前 GET /v1/models/hcolde/rvm 返回的 latest_version 仍为空，因此 POST /v1/models/hcolde/rvm/predictions 这条“按模型名直跑”的路径还不可用。
• 这是 Replicate 模型对象状态的问题，不是容器推理代码的问题；版本级调用和模型页访问已可用。