Repository for "Learning for Video Super-Resolution through HR Optical Flow Estimation", ACCV 2018
Video super-resolution (SR) aims to generate a sequence of high-resolution (HR) frames with plausible and temporally consistent details from their low-resolution (LR) counterparts. The generation of accurate correspondence plays a significant role in video SR. It is demonstrated by traditional video SR methods that simultaneous SR of both images and optical flows can provide accurate correspondences and better SR results. However, LR optical flows are used in existing deep learning based methods for correspondence generation. In this paper, we propose an end-to-end trainable video SR framework to super-resolve both images and optical flows. Specifically, we first propose an optical flow reconstruction network (OFRnet) to infer HR optical flows in a coarse-to-fine manner. Then, motion compensation is performed according to the HR optical flows. Finally, compensated LR inputs are fed to a super-resolution network (SRnet) to generate the SR results. Extensive experiments demonstrate that HR optical flows provide more accurate correspondences than their LR counterparts and improve both accuracy and consistency performance. Comparative results on the Vid4 and DAVIS-10 datasets show that our framework achieves the state-of-the-art performance.READ FULL TEXT VIEW PDF
Video super-resolution (SR) aims at generating a sequence of high-resolu...
Learning approaches have shown great success in the task of super-resolv...
The video super-resolution (VSR) task aims to restore a high-resolution ...
Stereo image pairs can be used to improve the performance of super-resol...
We study the video super-resolution (SR) problem not for visual quality,...
Motion boundary detection is a crucial yet challenging problem. Prior me...
Deep learning based methods have dominated super-resolution (SR) field d...
Repository for "Learning for Video Super-Resolution through HR Optical Flow Estimation", ACCV 2018
Super-resolution (SR) aims to generate high-resolution (HR) images or videos from their low-resolution (LR) counterparts. As a typical low-level computer vision problem, SR has been widely investigated for decades[23, 5, 7]. Recently, the prevalence of high-definition display further advances the development of SR. For single image SR, image details are recovered using the spatial correlation in a single frame. In contrast, inter-frame temporal correlation can further be exploited for video SR.
Since temporal correlation is crucial to video SR, the key to success lies in accurate correspondence generation. Numerous methods [6, 19, 22] have demonstrated that the correspondence generation and SR problems are closely interrelated and can boost each other’s accuracy. Therefore, these methods integrate the SR of both images and optical flows in a unified framework. However, current deep learning based methods [18, 13, 35, 2, 20, 21] mainly focus on the SR of images, and use LR optical flows to provide correspondences. Although LR optical flows can provide sub-pixel correspondences in LR images, their limited accuracy hinders the performance improvement for video SR, especially for scenarios with large upscaling factors.
In this paper, we propose an end-to-end trainable video SR framework to generate both HR images and optical flows. The SR of optical flows provides accurate correspondences, which not only improves the accuracy of each HR image, but also achieves better temporal consistency. We first introduce an optical flow reconstruction net (OFRnet) to reconstruct HR optical flows in a coarse-to-fine manner. These HR optical flows are then used to perform motion compensation on LR frames. A space-to-depth transformation is therefore used to bridge the resolution gap between HR optical flows and LR frames. Finally, the compensated LR frames are fed to a super-resolution net (SRnet) to generate each HR frame. Extensive evaluation is conducted to test our framework. Comparison to existing video SR methods shows that our framework achieves the state-of-the-art performance in terms of peak signal-to-noise ratio (PSNR) and structural similarity index (SSIM). Moreover, our framework achieves better temporal consistency for visual perception (as shown in Fig. 1).
Our main contributions can be summarized as follows: 1) We integrate the SR of both images and optical flows into a single SOF-VSR (super-resolving optical flow for video SR) network. The SR of optical flows provides accurate correspondences and improves the overall performance; 2) We propose an OFRnet to infer HR optical flows in a coarse-to-fine manner; 3) Extensive experiments have demonstrated the effectiveness of our framework. It is shown that our framework achieves the state-of-the-art performance.
In this section, we briefly review some major methods for single image SR and video SR.
Dong et al. 
proposed the pioneering work to use deep learning for single image SR. They used a three-layer convolutional neural network (CNN) to approximate the non-linear mapping from the LR image to the HR image. Recently, deeper and more complex network architectures have been proposed[14, 33, 11]. Kim et al. proposed a very deep super-resolution network (VDSR) with 20 convolutional layers. Tai et al.  developed a deep recursive residual network (DRRN) and used recursive learning to control the model parameters while increasing the depth. Hui et al.  proposed an information distillation network to reduce computational complexity and memory consumption.
Traditional video SR. To handle complex motion patterns in video sequences, Protter et al.  generalized the non-local means framework to address the SR problem. They used patch-wise spatio-temporal similarity to perform adaptive fusion of multiple frames. Takeda et al.  further introduced 3D kernel regression to exploit patch-wise spatio-temporal neighboring relationship. However, the resulting HR images of these two methods are over-smoothed. To exploit pixel-wise correspondences, optical flow is used in [6, 19, 22]. Since the accuracy of correspondences provided by optical flows in LR images is usually low , an iterative framework is used in these methods [6, 19, 22] to estimate both HR images and optical flows.
Deep video SR with separated motion compensation. Recently, deep learning has been investigated for video SR. Liao et al.  performed motion compensation under different parameter settings to generate an ensemble of SR-drafts, and then employed a CNN to recover high-frequency details from the ensemble. Kappelar et al.  also performed image alignment through optical flow estimation, and then passed the concatenation of compensated LR inputs to a CNN to reconstruct each HR frame. In these methods, motion compensation is separated from CNN. Therefore, it is difficult for them to obtain the overall optimal solution.
Deep video SR with integrated motion compensation. More recently, Caballero et al.  proposed the first end-to-end CNN framework (namely, VESPCN) for video SR. It comprises a motion compensation module and a sub-pixel convolutional layer used in . Since that, end-to-end framework with motion compensation dominates the research of video SR. Tao et al.  used the motion estimation module in VESPCN, and proposed an encode-decoder network based on LSTM. This architecture facilitates the extraction of temporal context. Liu et al.  customized ESPCN  to simultaneously process different numbers of LR frames. They then introduced a temporal adaptive network to aggregate multiple HR estimates with learned dynamic weights. Sajjadi et al.  proposed a frame-recurrent architecture to use previously inferred HR estimates for the SR of subsequent frames. The recurrent architecture can assimilate previous inferred HR frames without increase in computational demands.
It is already demonstrated by traditional video SR methods [6, 19, 22] that simultaneous SR of images and optical flows produces better result. However, current CNN-based methods only focus on the SR of images. Different from previous works, we propose an end-to-end video SR framework to super-resolve both images and optical flows. It is demonstrated that the SR of optical flows facilitates our framework to achieve the state-of-the-art performance.
Our framework takes consecutive LR frames as inputs and super-resolves the central frame. The LR inputs are first divided into pairs and fed to OFRnet to infer an HR optical flow. Then, a space-to-depth transformation  is employed to shuffle the HR optical flow into LR grids. Afterwards, motion compensation is performed to generate an LR draft cube. Finally, the draft cube is fed to SRnet to infer the HR frame. The overview of our framework is shown in Fig. 2.
It is demonstrated that CNN has the capability to learn the non-linear mapping between LR and HR images for the SR problem . Recent CNN-based works [4, 12] have also shown the potential for motion estimation. In this paper, we incorporate these two tasks into a unified network to infer HR optical flows from LR images. Specifically, our OFRnet takes a pair of LR frames and as inputs, and reconstruct an optical flow between their corresponding HR frames and :
where represents the HR optical flow and is the set of parameters.
Motivated by the pyramid optical flow estimation method in , we use a coarse-to-fine approach to handle complex motion patterns (especially large displacements). As illustrated in Fig. 3, a 3-level pyramid is employed in our OFRnet.
Level 1: The pair of LR images and are downsampled by a factor of 2 to produce and
, which are further concatenated and fed to a feature extraction layer. Then, two residual dense blocks (RDB)
with 4 layers and a growth rate of 32 are customized. Within each residual dense block, the first 3 layers are followed by a leaky ReLU using a leakage factor of 0.1, while the last layer performs feature fusion. The residual dense block works in a local residual learning manner with a local skip connection at the end. Once dense features are extracted by the residual dense blocks, they are concatenated and fed to a feature fusion layer. Then, the optical flowat this level is inferred by the subsequent flow estimation layer.
Level 2: Once the raw optical flow is obtained from level 1, it is upscaled by a factor of 2. The upscaled flow is then used to warp , resulting in . Next, , and are concatenated and fed to a network module. Note that, this module at level 2 is similar to that at level 1, except that residual learning is used.
Level 3: The module at level 2 generates an optical flow with the same size as the LR input . Therefore, the module at level 3 works as an SR part to infer the HR optical flow. The architecture at level 3 is similar to level 2 except that the flow estimation layer is replaced by a sub-pixel convolutional layer  for resolution enhancement.
Although numerous networks for SR [28, 16, 33, 11] and optical flow estimation [32, 27, 10] can be found in literature, our OFRnet is, to the best of our knowledge, the first unified network to integrate these two tasks. Note that, inferring HR optical flow from LR images is quite challenging, our OFRnet has demonstrated the potential of CNN to address this challenge. Our OFRnet is compact, with only 0.6M parameters. It is further demonstrated in Sec. 4.3 that the resulting HR optical flows benefit our video SR framework in both accuracy and consistency performance.
Once HR optical flows are produced by OFRnet, space-to-depth transformation is used to bridge the resolution gap between HR optical flows and LR frames. As illustrated in Fig. 4, regular LR grids are extracted from the HR flow and placed into the channel dimension to derive a flow cube with the same resolution as LR frames:
where and represent the size of the LR frame, is the upscaling factor. Note that, the magnitude of optical flow is divided by a scalar during the transformation to match the spatial resolution of LR frames.
Then, slices are extracted from the flow cube to warp the LR frame , resulting in multiple warped drafts:
where denotes warping operation and represents the warped drafts after concatenation, namely draft cube.
After motion compensation, all the drafts are concatenated with the central LR frame, as shown in Fig. 2. Then, the draft cube is fed to SRnet to infer the HR frame:
where is the super-resolved result of the central LR frame, represents the draft cube and is the set of parameters.
As illustrated in Fig. 5, the draft cube is first passed to a feature extraction layer with 64 kernels, and then the output features are fed to 5 residual dense blocks (which are similar to our OFRnet). Here, we increase the number of layers to 5 and the growth rate to 64 for each residual dense block. Afterwards, we concatenate all the outputs of residual dense blocks and use a feature fusion layer to distillate the dense features. Finally, a sub-pixel layer is used to generate the HR frame.
The combination of densely connected layers and residual learning in residual dense blocks has been demonstrated to have a contiguous memory mechanism [38, 9]. Therefore, we employ residual dense blocks in our SRnet to facilitate effective feature learning from preceding and current local features. Furthermore, feature reuse in the residual dense blocks improves the model compactness and stabilizes the training process.
We design two loss terms and for OFRnet and SRnet, respectively. For the training of OFRnet, intermediate supervision is used at each level of the pyramid:
here denotes the temporal window size and is the regularization term to constrain the smoothness of the optical flow. We empirically set and to make our OFRnet focus on the last level. We also set as the regularization coefficient.
For the training of SRnet, we use the widely applied mean square error (MSE) loss:
Finally, the total loss used for joint training is , where is empirically set to 0.01 to balance the two loss terms.
In this section, we first conduct ablation experiments to evaluate our framework. Then, we further compare our framework to several existing video SR methods.
We collected 152 1080P HD video clips from the CDVL Database111www.cdvl.org and the Ultra Video Group Database222ultravideo.cs.tut.fi. The collected videos cover diverse natural and urban scenes. We used 145 videos from the CDVL Database as the training set, and 7 videos from the Ultra Video Group Database as the validation set. Following the configuration in [19, 18, 35], we downsampled the video clips to the size of as the HR groundtruth using Matlab function. In this paper, we only focus on the upscaling factor of 4 since it is the most challenging case. Therefore, the HR video clips were further downsampled to produce LR inputs of size .
For fair comparison to the state-of-the-arts, we chose the widely used Vid4 benchmark dataset. We also used another 10 video clips from the DAVIS dataset  for further comparison, which we refer to as DAVIS-10.
|SOF-VSR w/o OFRnet||25.80||0.760||20.08||4.54||0.240|
|SOF-VSR w/o OFRnet||25.88||0.764||19.95||4.48||0.235|
|SOF-VSR w/o OFRnet + upsampling||25.86||0.763||19.92||4.50||0.231|
Following [3, 20], we converted input LR frames into YCbCR color space and only fed the luminance channel to our network. All metrics in this section are computed in the luminance channel. During the training phase, we randomly extracted 3 consecutive frames from an LR video clip, and randomly cropped a patch as the input. Meanwhile, its corresponding patch in HR video clip was cropped as groundtruth. Data augmentation was performed through rotation and reflection to improve the generalization ability of our network.
We implemented our framework in PyTorch. We applied the Adam solver with , and batch size of 16. The initial learning rate was set to and reduced to half after every 50K iterations. We trained our network from scratch for 300K iterations. All experiments were conducted on a PC with an Nvidia GTX 970 GPU.
In this section, we present ablation experiments on the Vid4 dataset to justify our design choices.
We proposed several variants of our SOF-VSR to perform ablation study. All the variants were re-trained for 300K iterations on the training data.
SOF-VSR w/o OFRnet. To handle complex motion patterns in video sequences, optical flow is used for motion compensation in our framework. To test the effectiveness of motion compensation for video SR, we removed the whole OFRnet and fed LR frames directly to our SRnet. Note that, replicated LR frames were used to match the dimension of the draft cube .
SOF-VSR w/o OFRnet. The SR of optical flows provides accurate correspondences for video SR and improves the overall performance. To validate the effectiveness of HR optical flows, we removed the module at level 3 in our OFRnet. Specifically, the LR optical flows at level 2 were directly used for motion compensation and subsequent processing. To match the dimension of the draft cube, compensated LR frames were also replicated before feeding to SRnet.
SOF-VSR w/o OFRnet + upsampling.
Super-resolving the optical flow can also be simply achieved using interpolation-based methods. However, our OFRnet can recover more reliable optical flow details. To demonstrate this, we removed the module at level 3 in our OFRnet, and upsampled the LR optical flows at level 2 using bilinear interpolation. Then, we used the modules in our original framework for subsequent processing.
To test the accuracy of individual output image, we used PSNR/SSIM as metrics. To further test the consistency performance, we used the temporal motion-based video integrity evaluation index (T-MOVIE) . Besides, we used MOVIE  and video quality measure with variable frame delay (VQM-VFD)  for overall evaluation. The MOVIE and VQM-VFD metrics are correlated with human perception and widely applied in video quality assessment. Evaluation results of our original framework and the 3 variants achieved on the Vid4 dataset are shown in Table 1.
Motion compensation. It can be observed from Table 1 that motion compensation plays a significant role in performance improvement. If OFRnet is removed, the PSNR/SSIM values are decreased from 26.01/0.771 to 25.80/0.760. Besides, the consistency performance is also degraded, with T-MOVIE value being increased to 20.08. That is because, it is difficult for SRnet to learn the non-linear mapping between LR and HR images under complex motion patterns.
HR optical flow. If modules at levels 1 and 2 are introduced to generate LR optical flows for motion compensation, the PSNR/SSIM values are increased to 25.88/0.764. However, the performance is still inferior to our SOF-VSR method using HR optical flows. That is because, HR optical flows provide more accurate correspondences for performance improvement. If bilinear interpolation is used to upsample LR optical flows, no consistent improvement can be observed. That is because, upsampling operation cannot recover reliable correspondence details as the module at level 3. To demonstrate this, we further compared the super-resolved optical flow (output at level 3), upsampled optical flow (upsampling result of the output at level 2) to the groundtruth. Since no groundtruth optical flow is available for the Vid4 dataset, we used the method proposed by Hu et al.  to compute the groundtruth optical flow. We used the average end-point error (EPE) for quantitative comparison, and present the results in Table 2.
It can be seen from Table 2 that the super-resolved optical flow significantly outperforms the upsampled optical flow, with an average EPE being reduced from 1.11 to 0.45. It demonstrates that the module at level 3 effectively recovers the correspondence details. Figure 6 further illustrates the qualitative comparison on City and Walk. In the upsampled optical flow, we can roughly distinguish the outlines of the building and the pedestrian. In contrast, more distinct edges can be observed in the super-resolved optical flow, with finer details being recovered. Although some checkboard artifacts generated by the sub-pixel layer can also be observed , the resulting HR optical flow provides highly accurate correspondences for the video SR task.
We first compared our framework to IDNnet  (the latest state-of-the-art single image SR method) and several video SR methods including VSRnet , VESCPN , DRVSR , TDVSR  and FRVSR  on the Vid4 dataset. Then, we conducted comparative experiments on the DAVIS-10 dataset.
|BI degradation model||BD degradation model|
|BI degradation model||BD degradation model|
For IDNnet and VSRnet, we used the codes provided by the authors to produce the results. For DRVSR and TDVSR, we used the output images provided by the authors. For VESCPN and FRVSR, the results reported in their papers [2, 29] are used. Here, we report the performance of FRVSR-3-64 since its network size is comparable to our SOF-VSR. Following , we crop borders of for fair comparison.
Note that, DRVSR and FRVSR are trained on a degradation model different from other networks. Specifically, the degradation model used in IDNnet, VSRnet, VESCPN and TDVSR is bicubic downsampling with Matlab function (denoted as BI). However, in DRVSR and FRVSR, the HR images are first blurred using Gaussian kernel and then downsampled by selecting every pixel (denoted as BD). Consequently, we re-trained our framework on the BD degradation model (denoted as SOF-VSR-BD) for fair comparison to DRVSR and FRVSR.
Without optimization of the implementation, our SOF-VSR network takes about 250ms to generate an HR image of size 720576 under configuration on an Nvidia GTX 970 GPU.
Evaluation on the Vid4 dataset. It can be observed from Table 3 that our SOF-VSR achieves the best performance for the BI degradation model in terms of all metrics. Specifically, the PSNR and SSIM values achieved by our framework are better than other methods by over 0.5 dB and 0.15 dB. That is because, more accurate correspondences can be provided by HR optical flows and therefore more reliable spatial details and temporal consistency can be well recovered.
For the BD degradation model, although the FRVSR-3-64 method achieves higher SSIM, our SOF-VSR-BD method outperforms FRVSR-3-64 in terms of PSNR. Compared to the DRVSR method, PSNR, SSIM and T-MOVIE values achieved by our SOF-VSRBD method are improved by a notable margin, while a comparable performance is achieved in terms of MOVIE and VQM-VFD.
We further show the trade-off between consistency and accuracy in Fig. 7. It can be seen that our SOF-VSR and SOF-VSR-BD methods achieve the highest PSNR values, while maintaining superior T-MOVIE performance.
Evaluation on the DAIVIS-10 dataset. It is clear in Table 4 that our SOF-VSR and SOF-VSR-BD methods surpass the state-of-the-arts for both the BI and BD degradation models in terms of all metrics. Since the DAVIS-10 dataset comprises scenes with fast moving objects, complex motion patterns (especially large displacements) lead to deterioration of existing video SR methods. In contrast, more accurate correspondences are provided by HR optical flows in our framework. Therefore, complex motion patterns can be handled more robustly and better performance can be achieved.
Figure 8 illustrates the qualitative results on two scenarios of the Vid4 dataset. It can be observed from the zoom-in regions that our framework recovers finer and more reliable details, such as the word “MAREE” and the stripes of the building. The qualitative comparison on the DAVIS-10 dataset (as shown in Fig. 9) also demonstrates the superior visual quality achieved by our framework. The pattern on the shorts, the word “PEUA” and the logo “CAT” are better recovered by our SOF-VSR and SOF-VSR-BD methods.
Figure 1 further shows the temporal profiles achieved on Calendar and City. It can be observed that the word “MAREE” can hardly be recognized by VSRnet in both image space and temporal profile. Although finer results are achieved by TDVSR, the building is still obviously distorted. In contrast, smooth and reliable patterns with fewer artifacts can be observed in temporal profiles of our results. In summary, our framework produces temporally more consistent results and better perceptual quality.
In this paper, we propose a deep end-to-end trainable video SR framework to super-resolve both images and optical flows. Our OFRnet first super-resolves the optical flows to provide accurate correspondences. Motion compensation is then performed based on HR optical flows and SRnet is used to infer the final results. Extensive experiments have demonstrated that our OFRnet can recover reliable correspondence details for the improvement of both accuracy and consistency performance. Comparison to existing video SR methods has shown that our framework achieves the state-of-the-art performance.