与EDVR相同。其中比较关键的是torch用1.1.0
pip install pillow==6.1 torch==1.1.0 torchvision==0.3.0
cd codes/dcn
python setup.py develop
教程:https://baijiahao.baidu.com/s?id=1660327134602942057&wfr=spider&for=pc
pip install ffmpeg-python
python gray.py --p 0.1 0.3 0.6 --input 视频路径 --Scratches 划痕素材路径 --output 输出路径
lmdb格式文件理论上可提高训练时数据的读取速度。
python create_lmdb.py --dataset ../datasets/Validation --output ../datasets/Validation --name Validation
示例
python main.py --save_folder ../experiments/experiment1 --train_dir /content/drive/Shared\ drives/furu.zhao/VFNet/datasets/Test2 --batchSize 2 --nFrames 5 --groups 8 --val_dir /content/drive/Shared\ drives/furu.zhao/VFNet/datasets/Validation --shuffle --pretrained True --pretrained_sr epoch_8.pth
(之后再补)
python test.py ..........