Nuke插件 – 视频抠图插件 RobustVideoMatting

Nuke 版本：Nuke12—–15

适用于 Foundry Nuke 的强大视频抠图插件

稳健视频抠图 (RVM) 专为稳健的人体视频抠图而设计。与将帧作为独立图像处理的现有神经模型不同，RVM 使用循环神经网络来处理具有时间记忆的视频。由于 RVM 使用 RNN，因此与使用 CNN 的 modnet 相比，它能提供更好的结果。CNN 必须单独处理每个帧，但 RNN 可以处理当前帧和周围的帧。

使用下面的说明，或观看上面的视频。

1. 将“KM_RVM_Nuke”文件夹复制到 nuke 插件路径（.nuke 文件夹）
2. 在 nuke 插件路径下的 init.py 文件中添加以下行：

nuke.pluginAddPath('./KM_RVM_Nuke')

重新启动 nuke，现在您可以通过“Keying”下的节点菜单找到 RVM 节点，或者在节点搜索栏中搜索“rvm”：

1. 下载并安装 MiniConda： https://repo.anaconda.com/miniconda/Miniconda3-latest-Windows-x86_64.exe

2. 打开位于“KM_RVM_Nuke”文件夹中的“Anaconda Prompt”，或在 Windows 搜索中找到它。逐行运行以下命令

conda create -n km_rvm python=3.9.0
conda activate km_rvm
pip install "av==8.0.3" "tqdm==4.61.1" "pims==0.5"
pip install torch torchvision

如果您想添加 gpu 支持，请用以下内容替换最后一行（它需要下载更多依赖项和大文件）：

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

，将“cu118”更改为您的显卡 Cuda 版本。Cuda 表如下，本页末尾。

一切设置完毕，您可以出发了！

如果您已经安装了自己的 Python，并且熟悉如何安装软件包，请安装依赖项。然后在 gizmo 的依赖项选项卡中，将 Python 路径设置为您的自定义 Python。

在“依赖项”选项卡中，单击“检查依赖项”按钮以显示您的系统配置。将其与要求进行比较。

通过键入 KM_RVM 创建 rvm 节点，或在 nuke 节点菜单中的键控类别下找到它。只需连接您的输入，或选择文件路径！如果您使用节点输入，它将首先进行 png seq 预合成。要直接输入文件，您需要将其作为 .mp4 或 png 序列。一旦工作完成，就会创建 roto matte 节点。输入预合成和结果将保存在您的 nuke 项目旁边的“KM_RVM_Data”文件夹中。块大小：并行处理的帧数。根据您的硬件使用此数字以获得更好的结果

使用此表找出你需要哪个版本的 pytorch cuda