本系统基于MSRN,可以实现图片和视频的超分辨率重建
python3.6
或以上环境
按照requirement.txt
文件安装所有需要的包
清空input
和output
文件夹
安装OpenH264解码器,下载链接
首先清空input文件夹,将图片文件复制进input文件夹中
然后运行demo.py
文件,其后缀格式如下:python demo.py <mode> --scale --device --format
mode
:选择模式,在图片重建时选择pic
--scale
:放大系数,可选择2、3、4,默认为2
--device
:使用设备,默认为cuda:0
--format
:输出格式,可选择bmp、jpg和png,若选择其他选项会报错,默认为png
例:
python demo.py pic --scale 3 --device cuda:0 --format jpg
首先清空input文件夹,将视频文件复制进input文件夹中,该系统会依次对每一个视频进行重建
然后运行demo.py
文件,其后缀格式如下:python demo.py <mode> --scale --device --format
mode
:选择模式,在视频重建时选择vid
--scale
:放大系数,默认为2
--device
:使用设备,默认为cuda:0
--format
:输出格式,可选择mp4和avi,若选择其他选项会报错,默认为mp4
例:
python demo.py vid --scale 2 --device cuda:0 --format mp4
输出文件将被保存在output
文件夹中
1、由于图像超分辨率重建较为吃性能,推荐选择显卡进行重建,同时显存最好要超过6GB,否则一些分辨率较高的图片或视频可能无法重建
2、机器内存推荐在8GB以上,若使用cpu进行重建的话推荐内存在16GB以上
3、由于OpenCV和算法的原因,输出图像可能会出现轻微色差,请见谅
4、input文件夹里不要把图片和视频混着放,会报错
5、视频超分辨率重建耗时巨大,请耐心等待