slowfast 代码复现 windows

slowfast 是一个基于 PyTorch 框架实现的视频分类模型。在 Windows 上进行 slowfast 代码的复现，需要先安装 PyTorch 和其他相关依赖。

以下是在 Windows 上复现 slowfast 的步骤：

1. 安装 Anaconda

在官网下载并安装 Anaconda，选择 Python 3.x 版本。

2. 创建虚拟环境

打开 Anaconda Prompt，输入以下命令创建一个名为 slowfast 的虚拟环境：

conda create -n slowfast python=3.7

激活虚拟环境：

conda activate slowfast

3. 安装依赖

安装 PyTorch 和其他相关依赖：

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
pip install cython matplotlib opencv-python-headless scipy termcolor yacs

如果没有 GPU，可以不安装 `cudatoolkit`。

4. 克隆代码

克隆 slowfast 代码：

git clone https://github.com/facebookresearch/SlowFast.git

进入 SlowFast 目录：

cd SlowFast

5. 编译运行库

在 SlowFast 目录下执行编译运行库：

cd slowfast
python setup.py build develop

6. 下载预训练模型

下载预训练模型，可以在 slowfast 官方 GitHub 上下载：

mkdir -p models/kinetics400
wget https://dl.fbaipublicfiles.com/slowfast/model_zoo/kinetics400/SLOWFAST_8x8_R50.pkl -O models/kinetics400/SLOWFAST_8x8_R50.pkl

7. 运行示例

在 SlowFast 目录下运行示例：

python tools/run_net.py \
  --cfg configs/Kinetics/c2/SLOWFAST_8x8_R50.yaml \
  DATA.PATH_TO_DATA_DIR path/to/dataset \
  TEST.CHECKPOINT_FILE_PATH models/kinetics400/SLOWFAST_8x8_R50.pkl

其中，`path/to/dataset` 为数据集路径。

至此，slowfast 在 Windows 上的代码复现完成。