提交 84313770 编写于 作者: L lijianshe02

add edvr inference code

上级 dad33770
...@@ -22,9 +22,7 @@ TRAIN: ...@@ -22,9 +22,7 @@ TRAIN:
number_frames: 5 number_frames: 5
batch_size: 32 batch_size: 32
file_root: "/workspace/video_test/video/data/dataset/edvr/REDS/train_sharp_bicubic/X4" file_root: "/workspace/video_test/video/data/dataset/edvr/REDS/train_sharp_bicubic/X4"
#file_root: "/ssd1/vis/sungaofeng/edvr/mmsr/datasets/REDS/train_sharp_bicubic/X4"
gt_root: "/workspace/video_test/video/data/dataset/edvr/REDS/train_sharp" gt_root: "/workspace/video_test/video/data/dataset/edvr/REDS/train_sharp"
#gt_root: "/ssd1/vis/sungaofeng/edvr/mmsr/datasets/REDS/train_sharp"
use_flip: True use_flip: True
use_rot: True use_rot: True
base_learning_rate: 0.0004 base_learning_rate: 0.0004
...@@ -57,8 +55,6 @@ TEST: ...@@ -57,8 +55,6 @@ TEST:
random_reverse: False random_reverse: False
number_frames: 5 number_frames: 5
batch_size: 1 batch_size: 1
#file_root: "/ssd1/vis/sungaofeng/edvr/mmsr/datasets/REDS4/sharp_bicubic"
#gt_root: "/ssd1/vis/sungaofeng/edvr/mmsr/datasets/REDS4/GT"
file_root: "/workspace/video_test/video/data/dataset/edvr/REDS4/sharp_bicubic" file_root: "/workspace/video_test/video/data/dataset/edvr/REDS4/sharp_bicubic"
gt_root: "/workspace/video_test/video/data/dataset/edvr/REDS4/GT" gt_root: "/workspace/video_test/video/data/dataset/edvr/REDS4/GT"
use_flip: False use_flip: False
...@@ -71,8 +67,6 @@ INFER: ...@@ -71,8 +67,6 @@ INFER:
random_reverse: False random_reverse: False
number_frames: 5 number_frames: 5
batch_size: 1 batch_size: 1
#file_root: "/workspace/video_test/video/data/dataset/edvr/REDS4/sharp_bicubic"
#gt_root: "/workspace/video_test/video/data/dataset/edvr/REDS4/GT"
file_root: "/workspace/color/input_frames" file_root: "/workspace/color/input_frames"
gt_root: "/workspace/video_test/video/data/dataset/edvr/REDS4/GT" gt_root: "/workspace/video_test/video/data/dataset/edvr/REDS4/GT"
use_flip: False use_flip: False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册