BasicVSR_pp video2frame_video2frame-frame2video

本文描述的实验是如何通过加载BasicVSR++模型实现video2frame

大家都会讲mmcv-full非常难安装，我之前也没有安装成功，后来更新了自己的驱动和CUDA版本，再pip install mmcv-full就莫名成功了，比较玄学。我的驱动控制面板以及CUDA如下所示：


总之就是版本都比较新了。
虽然大家都知道BasicVSR++，但是很多小伙伴并不关心模型如何搭建的，更想直接拿过来使用，所以我这里做了写了一点点，给有需要的同学做一个参考。
虽然没有用mmediting这个包，但是mmcv是一定要用的，因为要用到里面的可变形卷积。
文件目录的组织：

model.py用来存放模型的具体结构，utils将mmediting用到的函数都抽取出来放到这里，main.py是运行的主函数，目录result存放视频帧。

这里我将注册器给去掉了，因为我们自己直接做推理不要也可以。

utils.py里面存放了各种各样的函数工具，大部分是mmediting里面的，但是有两个裁剪输入的函数是我自己从其他地方加进来的，毕竟自己电脑只有4G的显存，不裁没法做呀，，，

主函数这样实现：

load_video用mmcv来读取视频，这样方便你后面的读图操作；inference和load_model是重点，需要多关注一下。

因为mmediting需要保存好的模型文件中，还会有其他的参数，比如迭代次数或者学习率一类的，我们不关心这些，我们只要网络里面的权重参数即可，所以要稍微处理一下再加载进来。

推理的时候就关心一下每次输入多少帧就可以了，因为我电脑是单卡，所以也没考虑用DataLoader再实现了，感觉没必要。
我的代码存放在这里了：https://github.com/ACALJJ32/Basicvsr_pp_video2frames.git