Next Article in Journal
Validation of the Activ8 Activity Monitor for Monitoring Postures, Motions, Transfers, and Steps of Hospitalized Patients
Next Article in Special Issue
Transformer-Based Multiple-Object Tracking via Anchor-Based-Query and Template Matching
Previous Article in Journal
RT-CLAD: Artificial Intelligence-Based Real-Time Chironomid Larva Detection in Drinking Water Treatment Plants
Previous Article in Special Issue
Inpainting with Separable Mask Update Convolution Network
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Edge-Oriented Compressed Video Super-Resolution

School of Telecommunications Engineering, Xidian University, Xi’an 710071, China
*
Author to whom correspondence should be addressed.
Sensors 2024, 24(1), 170; https://doi.org/10.3390/s24010170
Submission received: 20 November 2023 / Revised: 20 December 2023 / Accepted: 21 December 2023 / Published: 28 December 2023

Abstract

:
Due to the proliferation of video data in Internet of Things (IoT) systems, in order to reduce the data burden, most social media platforms typically employ downsampling to reduce the resolution of high-resolution (HR) videos before video coding. Consequently, the loss of detail and the introduction of additional artifacts seriously compromise the quality of experience (QoE). Recently, the task of compressive video super-resolution (CVSR) has garnered significant attention, aiming to simultaneously eliminate compression artifacts and enhance the resolution of compressed videos. In this paper, we propose an edge-oriented compressed video super-resolution network (EOCVSR), which focuses on reconstructing higher-quality details, to effectively address the CVSR task. Firstly, we devised a motion-guided alignment module (MGAM) to achieve precise bi-direction motion compensation in a multi-scale manner. Secondly, we introduced an edge-oriented recurrent block (EORB) to reconstruct edge information by combining the merits of explicit and implicit edge extraction. In addition, benefiting from the recurrent structure, the receptive field of EOCVSR can be enhanced and the features can be effectively refined without introducing additional parameters. Extensive experiments conducted on benchmark datasets demonstrate that our method surpasses the performance of state-of-the-art (SOTA) approaches in both quantitative and qualitative evaluations. Our approach can provide users with high-quality and cost-effective HR videos by integrating with sensors and codecs.

1. Introduction

Nowadays, the explosion of high-resolution (HR) videos has surged with the rapid advancement of portable device sensors and 5G communication technology. Some popular social media platforms, such as TikTok and WeChat, often downsample HR videos before compression to minimize the expenses associated with data storage and transmission, especially when bandwidth and storage space are severely constrained. The decoded low-resolution videos need to be upsampled back to their original resolution to meet user requirements. However, ensuring a satisfactory quality of experience (QoE) is extremely challenging due to the information loss caused by downsampling and the introduction of compression artifacts.
Despite the significant achievements of deep learning techniques in video quality enhancement (VQE) [1,2,3,4,5,6] and video super-resolution (VSR) tasks [7,8,9,10,11], simply cascading two networks to upsample the LR compressed videos in two stages often fail to yield satisfactory results. The main reason is that the information of the two independent processes cannot be effectively collaborated. Recently, there has been a growing interest in the task of compressed video super-resolution (CVSR) [12,13,14], which sought to address both compression artifacts removal and resolution enhancement in compressed videos simultaneously. An end-to-end restoration-reconstruction deep neural network (RR-DnCNN) [12] was first proposed to entirely solve degradation from compression and downsampling. The authors utilized upsampling skip connections to pass the useful features extracted by restoration to reconstruction. He et al. introduced a novel model, named Feature Multiplexing Video Super-Resolution for Compressed Video (FM-VSR) [13], to recover high-quality, high-resolution videos from low-resolution videos compressed with high compression rates. However, these methods lack targeted processing for high-frequency components, particularly edge components. Furthermore, they solely rely on the mean squared error (MSE) loss function during training, resulting in the inadequate reconstruction of details and overly smooth and blurry outputs. There is a strong demand for robust CVSR algorithms that collaborate with sensors and video codecs to produce exceptional HR videos.
To this end, we propose an edge-oriented compressed video super-resolution network (EOCVSR), which pays more attention to reconstructing high-quality details to effectively address the CVSR task. EOCVSR takes the target frame and its two temporally adjacent frames as inputs to fully leverage the information from multiple frames. Firstly, we devised a motion-guided alignment module (MGAM) to achieve precise bi-direction motion compensation. We analyzed the explicit optical flow between two frames in a multi-scale manner to generate an attention map. The offset of the deformable convolution [15] generated under the guidance of the attention map enables accurate implicit temporal alignment. Secondly, we proposed an edge-oriented recurrent block (EORB) to reconstruct edge information. We adopted several conventional filters and some learnable convolution kernels with specific shapes to derive edge information. By combining the merits of explicit and implicit edge extraction, we can perform a more targeted and high-quality reconstruction of high-frequency components. Furthermore, we designed a recurrent structure to enhance the receptive field and the performance of EOCVSR without introducing additional parameters. The edge-oriented loss function is also employed during training to boost performance. Extensive experiments conducted on benchmark datasets demonstrate that our method surpasses the performance of SOTA approaches in both quantitative and qualitative evaluations. The main contributions can be summarized as follows:
  • We propose an edge-oriented compressed video super-resolution network (EOCVSR) to address the CVSR problem. By incorporating a structure that specifically processes edge information and introduces edge-related loss functions, EOCVSR is able to reconstruct richer details and output higher-quality frames.
  • We propose a motion-guided alignment module (MGAM) to achieve precise bi-direction motion compensation. The utilization efficiency of temporal information is enhanced by employing explicit motion information to guide the generation of offsets for implicit temporal alignment.
  • We propose an edge-oriented recurrent block (EORB) to reconstruct edge information. Combining the merits of explicit and implicit edge extraction enables the high-quality reconstruction of high-frequency components. In addition, a recurrent structure is also adopted to realize effective feature refinement.

2. Related Works

2.1. Video Quality Enhancement (VQE)

With the widespread dissemination of video content and the increasing demand for storage, improving the quality of compressed videos has become a crucial task. Traditional video coding techniques introduce various distortions and compression artifacts during video compression, leading to a degradation in video quality. Consequently, researchers have begun exploring the use of deep learning methods to enhance the visual quality of compressed videos. An early work in the field of deep learning-based elimination of coding artifacts is the Variable-filter-size Residue-learning CNN (VRCNN) [1]. VRCNN integrates convolutional neural networks (CNNs) into the in-loop filter of High-Efficiency Video Coding (HEVC) [16] intra-coding, leading to improved coding performance. Wang et al. introduced a Deep CNN-based Auto-Decoder (DCAD) approach [2], which directly enhances the decoded frames without modifying the specific components of the decoder. Recognizing the distinction between intra-coding and inter-coding in HEVC, Yang et al. proposed the Quality-Enhancement Convolutional Neural Network (QECNN) [3]. QECNN comprises two subnetworks, namely QECNN-I and QECNN-B, which are designed to enhance the compressed I frames and P/B frames, respectively. Due to the lack of utilization of adjacent frames, the enhancement performance of these single-frame methods is limited. Therefore, some multi-frame methods [4,5,6] have been developed to leverage the temporal relationships between frames. Yang et al. observed that frames captured within a short period exhibit high similarity, leading them to propose a multi-frame quality enhancement (MFQE) approach [4]. This approach comprises a peak quality frames (PQFs) detector and a multi-frame convolutional neural network (MF-CNN). Building upon this, MFQE 2.0 [5] introduces an improved PQF-detector based on LSTM and a lightweight architecture for the MF-CNN, aiming to achieve enhanced performance more efficiently. Additionally, a spatio-temporal deformable fusion scheme [6] is employed, utilizing deformable convolutions to leverage temporal information. However, these existing methods need to be followed by an upsampling process to be applied to the CVSR task. The independence of the two processes from each other leads to an incoherent enhancement of the compressed video.

2.2. Video Super-Resolution (VSR)

Video super-resolution aims to improve video quality and detail reconstruction by learning the spatial and temporal characteristics of the video and upgrading the low-resolution video to high resolution. VSRnet [7] is an extension of the image super-resolution algorithm SRCNN [17] to the video domain. The major improvement lies in the inclusion of motion estimation and motion compensation modules, where the input is transformed from a single frame to multiple frames. Jo et al. introduced a DUF network [11] that generates dynamic upsampling filters and a residual image, which are computed depending on the local spatio-temporal neighborhood of each pixel to avoid explicit motion compensation. Tian et al. proposed a temporally deformable alignment network (TDAN) [10] to adaptively align the reference frame and each supporting frame with a feature level without computing optical flow. Concurrently, EDVR [8] devised an innovative spatio-temporal attention fusion module, along with enhanced deformable convolutions, to effectively handle motion compensation. However, when applying these methods to the CVSR task, compression artifacts may become more pronounced, significantly impacting video quality.

2.3. Compressed Video Super-Resolution (CVSR)

Recently, there has been a growing interest in the task of compressed video super-resolution (CVSR), which seeks to address both compression artifact removal and resolution enhancement in compressed videos simultaneously. An end-to-end restoration-reconstruction deep neural network (RR-DnCNN) [12] was proposed, which uses the degradation-aware technique to effectively deal with video compression distortion and upsampling degradation. In its extended version, RR-DnCNN v2 [14], the authors redesigned the network architecture by adopting a U-shaped form and incorporating upsampling skip connections. This architectural modification enables the propagation of valuable features captured during the restoration process to the subsequent reconstruction stage. He et al. proposed a one-stage video super-resolution network (FM-VSR) [13] to recover high-quality, high-resolution videos from low-resolution videos. They make full use of the internal correlation between the VQE and VSR tasks by adopting a feature multiplexing mechanism. However, these methods exhibit limitations in their treatment of high-frequency components, specifically edge components, as they do not incorporate targeted processing techniques. Additionally, their reliance solely on the mean squared error (MSE) loss function during the training phase leads to suboptimal reconstruction of fine details and outputs that are excessively smooth and blurry.

3. The Proposed EOCVSR Approach

3.1. Overall Framework

The overall framework of our proposed EOCVSR approach is depicted in Figure 1. EOCVSR takes the target frame I t and its two temporally adjacent frames I t 1 and I t + 1 as the input. First of all, a parameter-shared feature extraction module transforms three input frames from pixel space to feature space. As shown in Figure 1, the following procedure can be divided into two stages: restoration and reconstruction. The restoration stage aims to eliminate the compression artifacts. To make full use of temporal multi-frame information, a motion-guided alignment module is employed to achieve precise bi-direction motion compensation. Then, we stack three edge-oriented recurrent blocks (EORBs) to refine the aligned feature. The intermediate outputs of EORBs in the restoration stage are fused by a convolution layer. The fused feature is fed into a feature reconstruction module to obtain a residual image, which will be added to I t to generate the result of the restoration stage O t r e s .
The reconstruction stage, which takes the fused feature output by the restoration stage as the input, aims to enhance the resolution and reconstruct high-quality details. The pixel-shuffle operation is first adopted to upsample the input feature. Another three cascading EORBs extract edge information at a larger scale. In the reconstruction stage, the output of each EORB is integrated with the output of the EORB, which is in the same order in the restoration stage. Such internal correlation between the two stages improves the learning ability of EOCVSR. Same as in the restoration stage, we use a convolution layer to fuse the outputs of EORBs from the reconstruction stage. Finally, the result of the reconstruction stage O t r e c can be obtained by adding the residual image reconstructed from the high-resolution fused feature to the result of upsampling I t using the Bi-cubic operation. The function of our proposed EOCVSR can be formulated as
O t r e s , O t r e c = Φ ( I t 1 , I t , I t + 1 | θ ) ,
where Φ is the proposed approach and θ is the parameters of the network. The symbols that will appear with the corresponding explanations are shown in Table 1.

3.2. Feature Extraction Module

The feature extraction module is defined as
F n = f f e ( I n ) , n { t 1 , t , t + 1 } ,
where f f e ( · ) denotes the feature extraction function, and F represents the extracted feature of the input frames I. First, a convolution layer (Conv) is used to transform the input frames from a pixel space to a higher dimensional feature space. Then, we adopt a typical residual block (RB) [18] to roughly process the features. The parameters used to process the inputted three frames are shared.

3.3. Motion-Guided Alignment Module

The motion-guided alignment module (MGAM) aims to achieve precise bi-direction motion compensation and derive useful temporal multi-frame information for restoration. As shown in Figure 2, the structure of MGAM is divided into two parts, bi-direction motion-guided alignment and feature modulation. In the first part, we utilize explicit motion prior, i.e., optical flow, to guide feature-level temporal alignment. Take the forward alignment as an example, we first obtain the motion vector M V t 1 t using a pre-trained SpyNet [19], which is a widely used optical flow prediction network.
M V t 1 t = S p y N e t ( I t , I t 1 )
Then, we employ a parameter-shared Conv with a different dilation d { 1 , 2 , 4 } to implement multi-scale analysis on M V t 1 t . The analyzed results are concatenated, fused, and outputted by a softmax layer, resulting in the generation of a motion prior in the form of an attention map.
h 1 = C o n v ( M V t 1 t ) d = 1 , h 2 = C o n v ( M V t 1 t ) d = 2 , h 3 = C o n v ( M V t 1 t ) d = 3 , M o t i o n P r i o r = S o f t m a x ( C o n v ( [ h 1 , h 2 , h 3 ] ) ) . ,
where [ · , · ] and S o f t m a x are concatenation and the softmax layer. Meanwhile, an offset of the deformable convolution (DCN) kernel O t 1 t is dynamically predicted from F t and F t 1 . Before directly applying O t 1 t to DCN, we fine-tune O t 1 t by multiplying the motion prior with it, so that it can better capture the motion correlation. By feeding F t 1 and the fine-tuned O t 1 t into a DCN, a forward-aligned feature F t 1 t a l i g n e d can be obtained. The detailed process is given below:
O t t 1 = C o n v ( [ F t , F t 1 ] ) , F t 1 t a l i g n e d = D C N ( F t 1 , O t t 1 M o t i o n P r i o r ) . ,
where D C N and ⊙ denote the deformable convolution layer and element-wise product. The backward alignment takes F t , F t + 1 , and the backward optical flow takes M V t + 1 t as the input and outputs a backward-aligned feature F t + 1 t a l i g n e d . The processing is mirrored in the forward alignment.
In the second part, we adopt the method of feature modulation to realize the fusion of multi-frame information. F t 1 t a l i g n e d and F t + 1 t a l i g n e d are first concatenated together, and then the concatenation result is fed into two Convs to generate the shift and add weights for modulation, respectively. This operation can be described as
s h i f t = C o n v ( [ F t 1 t a l i g n e d , F t + 1 t a l i g n e d ] ) , a d d = C o n v ( [ F t 1 t a l i g n e d , F t + 1 t a l i g n e d ] ) , F t m g a m = F t s h i f t + a d d . .
Thanks to precise motion alignment in the MGAM, the aligned features provide significant improvement to the overall performance.

3.4. Edge-Oriented Recurrent Block

Most existing CVSR approaches lack targeted processing for high-frequency components, particularly edge components. Furthermore, they solely rely on mean squared error (MSE) loss function during training, resulting in inadequate reconstruction of details and overly smooth and blurry outputs. To this end, we devise an edge-orient recurrent block (EORB) to augment the network’s capacity for perceiving and reconstructing details. As shown in Figure 3, the recurrent unit is the key component of the EORB. The input of the recurrent unit is first processed by six different filters for edge-aware. To perceive horizontal edges, we adopt a horizontal Sobel filter and a learnable Conv with the size of 1 × 9 . To perceive vertical edges, we adopt a vertical Sobel filter and another learnable 9 × 1 Conv. We also extract the second-order spatial derivative using a Laplacian filter, collaborating with a learnable 3 × 3 Conv. Combining the merits of pre-defined and learnable edge filters, the edge information can be efficiently derived. It is worth noting that the shapes of the six detectors will not change, and the parameters of the three learnable edge detectors can be updated through the back-propagation operation. Then, the summation of all the outputs, followed by the application of a global average pooling (GAP) layer, two Convs, and a softmax layer, generates the weights corresponding to each output. By multiplying the outputs with their corresponding weights and subsequently accumulating them, valuable edge information is filtered and preserved. The process within the recurrent unit is described as
O 1 = C o n v 9 × 1 ( F t R U k ) , O 2 = S o b e l D x ( F t R U k ) O 3 = C o n v 1 × 9 ( F t R U k ) , O 4 = S o b e l D y ( F t R U k ) O 5 = C o n v 3 × 3 ( F t R U k ) , O 6 = L a p l a c i a n ( F t R U k ) W 1 W 6 = S o f t m a x ( C o n v ( C o n v ( G A P ( i = 1 6 O i ) ) ) ) , R U ( · ) = C o n v ( i = 1 6 O i W i ) ,
where R U ( · ) represents the process of the recurrent unit. k [ 1 , K 1 ] stands for the current iteration of recursion and K is the maximum number of recursions. At last, we stack three RBs to further process the feature. The output of the previous RB is utilized as the input of the next RB. Only the output of the last RB is added to the input feature of the EORB, which is F t m g a m .
To enhance the receptive field and performance without introducing additional parameters, we adopt a recurrent strategy. For the current recursion, the output of the previous recursion is added to the input of the EORB F t i n p u t and is subsequently passed through a channel attention layer. No matter how many times the recursion is performed during the training, it still belongs to the forward inference phase and does not involve updating the parameters. Therefore, in each recursion, the parameters of the recurrent unit are shared. The output of the EORB F t e o r b is obtained by using a Conv to fuse the concatenation result of the outputs of all recursions. The detailed process is given below:
F t R U k = C A ( R U ( F t i n p u t ) + F t i n p u t ) , F t R U ( k + 1 ) = C A ( R U ( F t R U k ) + F t i n p u t ) F t e o r b = C o n v ( [ F t R U 1 , F t R U ( k + 1 ) ] ) . .

3.5. Feature Reconstruction Module

The feature reconstruction module takes the fused result of the outputs of all previous EORBs as the input. The feature reconstruction module consists of an RB and a Conv, which outputs a residual image R t . The process is defined as
R t = f f r ( C o n v ( [ F t e o r b M ] ) ) ,
where f f r ( · ) is the feature reconstruction function. M is the number of EORBs of each stage. In the restoration stage, the residual image will be added to I t to obtain O t r e s . As for the reconstruction stage, the residual image will be added to the upsampling result of I t to obtain O t r e c . It is worth mentioning that the feature reconstruction modules in the restoration and reconstruction stages use different parameters.

3.6. Loss Function

Firstly, in addition to calculating the reconstruction error between the super-resolution result O t r e c and the uncompressed HR frame G t H R , we also use the uncompressed LR frame G t L R as an auxiliary supervised label to assist the network training. Thus, the total loss function is defined as
L t o t a l = α L r e s + L r e c ,
where α is the weighted factor for the restoration loss.
However, only relying on the mean squared error (MSE) loss function during training may cause inadequate reconstruction of details and overly smooth and blurry outputs. To this end, we introduce an edge-orient loss function to direct constraints on the learning of edge reconstruction. Specifically, we explicitly extract high-frequency components using a Gaussian kernel blur g ( · , · ) with the width of σ . The details of the loss function are shown below:
L r e s = L m s e ( O t r e s , G t L R ) + β L m s e ( O E t r e s , G E t L R ) L r e c = L m s e ( O t r e c , G t H R ) + β L m s e ( O E t r e c , G E t H R ) O E t = O t g ( O t , σ = 3 ) G E t = G t g ( G t , σ = 3 ) .

4. Results

4.1. Experimental Setup

To train our proposed EOCVSR, we use the Vimeo dataset [19], which contains about 65,000 video sequences. Each sequence has seven frames with 448 × 256 resolution. We first use Bi-cubic interpolation to obtain uncompressed LR videos by a downsampling factor of two. Then, we compress these LR videos using FFmpeg [20] with the default mode and CRFs = 32, 37, 42, and 47. Finally, the bitstream is decoded to generate the compressed LR videos. To evaluate the trained EOCVSR, we adopt the test sequences from the standard HEVC common test condition (CTC) [21], the UVG dataset [22], and the MCL-JCV dataset [23]. The downsampling and compression settings are consistent with those in the training.
During the training, we randomly crop 120 × 120 patches from a mini-batch as the input. The batch size is set as eight. The hyperparameters regarding the network structure K and M are both set as three. The model trained with the loss function described in Section 3.5, and the weights α and β , are set as 0.2 and 0.1, respectively. The learning rate is initialized as 1 × 10 4 and then divided by a factor of 10 every 30 epochs. The training stops after 100 epochs. The Adam optimizer [23] is used by setting β 1 = 0.9 and β 2 = 0.999 . During the evaluation, we use BD-BR [24], which presents the quality improvement (dB) at the same bitrate, and PSNR for quantitative analysis of the compressed video super-resolution results. All the models are implemented with PyTorch 1.4.0 [25] on Nvidia Geforce 2080Ti GPUs. All calculations are on the luminance channel (Y channel).

4.2. Performance of Proposed EOCVSR

The performance of our proposed EOCVSR is compared with the latest CVSR approaches, including RR-DnCNN v2 [14] and FM-VSR [13]. As shown in Table 2, we compare the characteristics between EOCVSR and other CVSR approaches. We also retrain some advanced video super-resolution approaches, including [8] and BasicVSR++ [9], over our training dataset for comparison. The comparison of the size of each model and computation cost is displayed in Table 3. We evaluate the quality enhancement, the rate–distortion performance, and subjective performance. The details are described as follows.

4.2.1. Quality Enhancement

Table 4 presents the PSNR results over HEVC standard test sequences. First, the results illustrate that the proposed EOCVSR outperforms all the prior approaches over four compression ratios. Specifically, the PSNR of EOCVSR is 29.037 dB, which is 0.038 dB higher than the state-of-the-art approach, i.e., BasicVSR++ (28.999 dB), and 0.107–0.325 dB higher than others [8,13,14]. Table 5 shows the PSNR results over the UVG and MCL-JCV datasets, and the results demonstrate that EOCVSR also achieves the most significant boost. In terms of the UVG dataset, the PSNR improvement of EOCVSR ranges from 0.022 to 0.140. In terms of the MCL-JCV dataset, the PSNR improvement of EOCVSR ranges from 0.031 to 0.200. Thanks to MGAM’s precise motion alignment and EORM’s powerful edge information extraction and reconstruction capabilities, our proposed EOCVSR achieves the highest quality enhancement over all evaluation datasets.

4.2.2. Rate–Distortion Performance

Here, we evaluate the rate–distortion of EOCVSR over three datasets, and the results are shown in Table 6. Considering full-resolution compression as an anchor, the performance has been improved in terms of 14.364% BD-BR reduction on average over HEVC standard test sequences, which is 0.843% more than BasicVSR++ (13.521%), and 2.737–8.277% more than other approaches [8,13,14]. As for the UVG and the MCL-JCV datasets, the bitrate saving achieves 29.849% and 28.217%. To present more intuitively, rate–distortion curves of our and other approaches over all test sequences are shown in Figure 4. In this figure, we can observe that the curve of EOCVSR is above that of full-resolution compression. In brief, the quantitative results demonstrate that our proposed EOCVSR realizes the best compression performance.

4.2.3. Subjective Performance

Figure 5 shows the visual results of the different methods. Observing the results, it is evident that our proposed EOCVSR has demonstrated substantial advancements in visual perception when compared to other approaches. The compression artifacts are effectively eliminated. Thanks to the excellent edge-awareness of EOCVSR, the edges and textures are reconstructed with high quality. For example, the face in the sequence BasketballDrive and the letters in the sequence KristenAndSara. Therefore, our EOCVSR approach achieves promising performance in subjective quality.

4.3. Ablation Study

In this section, we conduct several ablation experiments to analyze the impact of different structures of the proposed EOCVSR on performance. All the experiments are performed over the MCL-JCV dataset and the compression ratio CRF is set as 47. In general, the results demonstrate that EOCVSR is delicately designed to achieve optimal performance. The details are described as follows.

4.3.1. Analysis of the EORB

An ablation study has been conducted to verify the effectiveness of our proposed EORB. We modify the EORB by removing the six filters used to perceive the edges and leaving only the RBs. As shown in Table 7, the original EORB achieves a 28.386 dB improvement in PSNR, while the performance of the modified EORB degrades to 28.317 dB. The results prove that combining the merits of pre-defined and learnable edge filters enables EOCVSR to efficiently extract the edge information for higher performance. Furthermore, we provide a feature map visualization example. As shown in Figure 6, the original EORB is more sensitive to edges and textures.

4.3.2. Analysis of the Number of Recursions K

We design a recurrent structure to enhance the receptive field and the performance of EOCVSR without introducing additional parameters, and the number of recursions within an EORB has an impact on the performance of EOCVSR. As shown in Figure 7a, the performance increases quickly when K is from one to three, while the performance increases slowly when K is larger than three. Considering more recursions may extend the processing time, K is set as three in this work.

4.3.3. Analysis of the Number of EORBs M

We evaluate EOCVSR with different numbers of EORBs. Integrating more EORBs can enhance the representation capabilities of the network but introduce more parameters. Figure 7b shows the performance of quality enhancement versus the number of EORBs M. The performance gain increases slowly when M is from 6 to 10. Thus, M is set as six (three in the restoration stage and another three in the reconstruction stage) in this work.

4.3.4. Model Adaption

We evaluate the adaptability of the proposed EOCVSR to different types of tasks. Theoretically, the proposed design can be applied to many other video tasks, such as denoising tasks, enhancement tasks, deblurring tasks, and so on. We add Gaussian noise to the image frames to train and test the ability of EOCVSR to perform super-resolution denoising at noise levels of σ = 15, 25, and 50. The parameter settings and network training are the same as for the original task. We compared the performance of EOCVSR with MIRNet [26]. As shown in Table 8, the PSNR of EOCVSR is improved by 0.45–1.4 dB. The results show that our method outperforms the learning-based super-resolution denoising method, illustrating that our model possesses the ability to adapt to different types of tasks.

5. Conclusions

In this paper, we proposed an edge-oriented compressed video super-resolution network (EOCVSR), which pays more attention to reconstructing high-quality details to effectively address the CVSR task. We devised a motion-guided alignment module (MGAM), which uses the explicit optical flow to direct the generation of offsets for deformable convolutions leading to precise bi-direction motion compensation. In addition, we proposed an edge-oriented recurrent block (EORB) to reconstruct edge information. Several pre-defined filters and some learnable convolution kernels with specific shapes were exploited to derive edge information. By combining the merits of explicit and implicit edge extraction, we can perform a more targeted and high-quality reconstruction of high-frequency components. We also designed a recurrent structure to enhance the receptive field and the performance of EOCVSR without introducing additional parameters. Our approach can provide users with high-quality and cost-effective HR videos by integrating with sensors and codecs.

Author Contributions

Conceptualization, Z.W. and G.H.; methodology, Z.W.; software, Z.W.; validation, G.Q.; formal analysis, Z.W.; investigation, G.H.; resources, G.H.; data curation, G.Q.; writing—original draft preparation, Z.W. and G.Q.; writing—review and editing, G.H.; visualization, Z.W. and G.Q.; supervision, G.H.; project administration, G.H.; funding acquisition, G.H. All authors have read and agreed to the published version of this manuscript.

Funding

This work is supported by project CEIEC-2022-ZM02-0247.

Institutional Review Board Statement

Not applicable.

Informed Consent Statement

Not applicable.

Data Availability Statement

The data presented in this study are available on request from the corresponding author.

Conflicts of Interest

The authors declare no conflicts of interest.

References

  1. Dai, Y.; Liu, D.; Wu, F. A convolutional neural network approach for post-processing in HEVC intra coding. In Proceedings of the MultiMedia Modeling: 23rd International Conference, MMM 2017, Reykjavik, Iceland, 4–6 January 2017; Proceedings, Part I 23. Springer: Berlin/Heidelberg, Germany, 2017; pp. 28–39. [Google Scholar]
  2. Wang, T.; Chen, M.; Chao, H. A novel deep learning-based method of improving coding efficiency from the decoder-end for HEVC. In Proceedings of the 2017 Data Compression Conference (DCC), Snowbird, UT, USA, 4–7 April 2017; pp. 410–419. [Google Scholar]
  3. Pan, Z.; Yi, X.; Zhang, Y.; Jeon, B.; Kwong, S. Efficient in-loop filtering based on enhanced deep convolutional neural networks for HEVC. IEEE Trans. Image Process. 2020, 29, 5352–5366. [Google Scholar] [CrossRef] [PubMed]
  4. Yang, R.; Xu, M.; Wang, Z.; Li, T. Multi-frame quality enhancement for compressed video. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA, 18–23 June 2018; pp. 6664–6673. [Google Scholar]
  5. Guan, Z.; Xing, Q.; Xu, M.; Yang, R.; Liu, T.; Wang, Z. MFQE 2.0: A new approach for multi-frame quality enhancement on compressed video. IEEE Trans. Pattern Anal. Mach. Intell. 2019, 43, 949–963. [Google Scholar] [CrossRef] [PubMed]
  6. Deng, J.; Wang, L.; Pu, S.; Zhuo, C. Spatio-temporal deformable convolution for compressed video quality enhancement. In Proceedings of the AAAI Conference on Artificial Intelligence, New York, NY, USA, 7–12 February 2020; Volume 34, pp. 10696–10703. [Google Scholar]
  7. Kappeler, A.; Yoo, S.; Dai, Q.; Katsaggelos, A.K. Video super-resolution with convolutional neural networks. IEEE Trans. Comput. Imaging 2016, 2, 109–122. [Google Scholar] [CrossRef]
  8. Wang, X.; Chan, K.C.; Yu, K.; Dong, C.; Change Loy, C. Edvr: Video restoration with enhanced deformable convolutional networks. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, Long Beach, CA, USA, 16–20 June 2019. [Google Scholar]
  9. Chan, K.C.; Zhou, S.; Xu, X.; Loy, C.C. Basicvsr++: Improving video super-resolution with enhanced propagation and alignment. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, New Orleans, LA, USA, 18–24 June 2022; pp. 5972–5981. [Google Scholar]
  10. Tian, Y.; Zhang, Y.; Fu, Y.; Xu, C. Tdan: Temporally-deformable alignment network for video super-resolution. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA, 14–19 June 2020; pp. 3360–3369. [Google Scholar]
  11. Jo, Y.; Oh, S.W.; Kang, J.; Kim, S.J. Deep video super-resolution network using dynamic upsampling filters without explicit motion compensation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA, 18–23 June 2018; pp. 3224–3232. [Google Scholar]
  12. Ho, M.M.; He, G.; Wang, Z.; Zhou, J. Down-sampling based video coding with degradation-aware restoration-reconstruction deep neural network. In Proceedings of the MultiMedia Modeling: 26th International Conference, MMM 2020, Daejeon, Republic of Korea, 5–8 January 2020; Proceedings, Part I 26. Springer: Berlin/Heidelberg, Germany, 2020; pp. 99–110. [Google Scholar]
  13. He, G.; Wu, S.; Pei, S.; Xu, L.; Wu, C.; Xu, K.; Li, Y. FM-VSR: Feature Multiplexing Video Super-Resolution for Compressed Video. IEEE Access 2021, 9, 88060–88068. [Google Scholar] [CrossRef]
  14. Ho, M.M.; Zhou, J.; He, G. RR-DnCNN v2. 0: Enhanced restoration-reconstruction deep neural network for down-sampling-based video coding. IEEE Trans. Image Process. 2021, 30, 1702–1715. [Google Scholar] [CrossRef]
  15. Dai, J.; Qi, H.; Xiong, Y.; Li, Y.; Zhang, G.; Hu, H.; Wei, Y. Deformable convolutional networks. In Proceedings of the IEEE International Vonference on Vomputer Vision, Venice, Italy, 22–29 October 2017; pp. 764–773. [Google Scholar]
  16. Sullivan, G.J.; Ohm, J.R.; Han, W.J.; Wiegand, T. Overview of the high efficiency video coding (HEVC) standard. IEEE Trans. Circuits Syst. Video Technol. 2012, 22, 1649–1668. [Google Scholar] [CrossRef]
  17. Dong, C.; Loy, C.C.; Tang, X. Accelerating the super-resolution convolutional neural network. In Proceedings of the Computer Vision–ECCV 2016: 14th European Conference, Amsterdam, The Netherlands, 11–14 October 2016; Proceedings, Part II 14. Springer: Berlin/Heidelberg, Germany, 2016; pp. 391–407. [Google Scholar]
  18. Lim, B.; Son, S.; Kim, H.; Nah, S.; Mu Lee, K. Enhanced deep residual networks for single image super-resolution. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition Workshops, Honolulu, HI, USA, 21–26 July 2017; pp. 136–144. [Google Scholar]
  19. Ranjan, A.; Black, M.J. Optical flow estimation using a spatial pyramid network. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA, 21–26 July 2017; pp. 4161–4170. [Google Scholar]
  20. Newmarch, J.; Newmarch, J. Ffmpeg/libav. Linux Sound Program. 2017, 19, 227–234. [Google Scholar]
  21. Bossen, F. Common test conditions and software reference configurations. JCTVC-L1100 2013, 12, 1. [Google Scholar]
  22. Mercat, A.; Viitanen, M.; Vanne, J. UVG dataset: 50/120fps 4K sequences for video codec analysis and development. In Proceedings of the 11th ACM Multimedia Systems Conference, Istanbul, Turkey, 8–11 June 2020; pp. 297–302. [Google Scholar]
  23. Wang, H.; Gan, W.; Hu, S.; Lin, J.Y.; Jin, L.; Song, L.; Wang, P.; Katsavounidis, I.; Aaron, A.; Kuo, C.C.J. MCL-JCV: A JND-based H. 264/AVC video quality assessment dataset. In Proceedings of the 2016 IEEE International Conference on Image Processing (ICIP), Phoenix, AZ, USA, 25–28 September 2016; pp. 1509–1513. [Google Scholar]
  24. Grois, D.; Marpe, D.; Mulayoff, A.; Itzhaky, B.; Hadar, O. Performance comparison of h. 265/mpeg-hevc, vp9, and h. 264/mpeg-avc encoders. In Proceedings of the 2013 Picture Coding Symposium (PCS), San Jose, CA, USA, 8–11 December 2013; pp. 394–397. [Google Scholar]
  25. Imambi, S.; Prakash, K.B.; Kanagachidambaresan, G. PyTorch. In Programming with TensorFlow: Solution for Edge Computing Applications; Springer: Berlin/Heidelberg, Germany, 2021; pp. 87–104. [Google Scholar]
  26. Zamir, S.W.; Arora, A.; Khan, S.; Hayat, M.; Khan, F.S.; Yang, M.H.; Shao, L. Learning Enriched Features for Real Image Restoration and Enhancement. arXiv 2020, arXiv:2003.06792v2. [Google Scholar]
Figure 1. The overall framework of our proposed EOCVSR.
Figure 1. The overall framework of our proposed EOCVSR.
Sensors 24 00170 g001
Figure 2. The structure of motion-guided alignment module (MGAM).
Figure 2. The structure of motion-guided alignment module (MGAM).
Sensors 24 00170 g002
Figure 3. The structure of edge-oriented recurrent block (EORB).
Figure 3. The structure of edge-oriented recurrent block (EORB).
Sensors 24 00170 g003
Figure 4. The RD curve over all test datasets of various resolutions under different compression distortion conditions.
Figure 4. The RD curve over all test datasets of various resolutions under different compression distortion conditions.
Sensors 24 00170 g004
Figure 5. Visual comparisons of different methods on video frames from HEVC standard test sequences (QP = 42). The zoom-in of red box area is shown.
Figure 5. Visual comparisons of different methods on video frames from HEVC standard test sequences (QP = 42). The zoom-in of red box area is shown.
Sensors 24 00170 g005
Figure 6. The feature map output by EORB with edge-perceiving filters vs. without edge-perceiving filters.
Figure 6. The feature map output by EORB with edge-perceiving filters vs. without edge-perceiving filters.
Sensors 24 00170 g006
Figure 7. Ablation study on (a) the number of recursions within an EORB and (b) the number of EORBs.
Figure 7. Ablation study on (a) the number of recursions within an EORB and (b) the number of EORBs.
Sensors 24 00170 g007
Table 1. List of symbols.
Table 1. List of symbols.
SymbolsExplanation
I t Input video frame t
O t r e s Video restoration output
O t r e c Video reconstruction output
F t The feature of frame t
M V Motion vector
OThe offset of the deformable convolution kernel
F a l i g n e d The aligned feature
F t m g a m The output feature of MGAM for frame t
C o n v m × n Convolutional layer with kernel of m × n
S o f t m a x Softmax normalized activation function
S o b e l D x Vertical edge detection operator of Sobel
S o b e l D y Horizontal edge detection operator of Sobel
L a p l a c i a n Laplacian edge detection operator
Element-wise product
Element-wise add
G t L R Low-resolution Ground Truth at frame t
G t H R High-resolution Ground Truth at frame t
L m s e Mean squared error loss function
Table 2. Comparison with previous CVSR works.
Table 2. Comparison with previous CVSR works.
ApproachMulti-Frame UtilizationEdge Preservation
RR-DnCNN v2××
FM-VSR×
EOCVSR (proposed)
Table 3. The comparison of the number of model parameters and GFLOPs. GFLOPs are calculated on an image with an input size of 64 × 64.
Table 3. The comparison of the number of model parameters and GFLOPs. GFLOPs are calculated on an image with an input size of 64 × 64.
RR-DnCNN v2FM-VSREDVRBasicVSR++EOCVSR
Parameter Number1.8M7.1M2.7M7.1M3.5M
GFLOPs20.996.166.6104.388.4
Table 4. The comparison of PSNR gain over HEVC standard test sequences. Red indicates the best performance, and blue indicates the second-best.
Table 4. The comparison of PSNR gain over HEVC standard test sequences. Red indicates the best performance, and blue indicates the second-best.
QPClassSequencesRR-DnCNN v2FM-VSREDVRBasicVSR++EOCVSR
32APeopleOnStreet29.60829.81328.85930.02930.129
Traffic32.81732.86632.97633.06933.102
BBasketballDrive31.72631.90232.42232.47132.531
Cactus31.01231.29731.56631.71231.735
Kimono34.45634.73534.79934.84434.876
ParkScene31.22931.38431.60831.65531.671
CBasketballDrill30.17730.37930.46030.63630.706
RaceHorses24.54426.11127.52427.59927.609
EFourPeople33.75733.86533.92934.10334.260
Johnny35.70535.92636.10136.13736.162
KristenAndSara34.28234.51034.66034.91934.973
Average32.01732.07232.34832.47032.523
37Average29.96230.09330.13730.21030.262
42Average27.55927.61027.77527.85127.838
47Average25.30825.36625.45925.46425.524
Overall28.71228.78528.93028.99929.037
Table 5. The comparison of average PSNR gain over the UVG and MCL-JCV datasets. Red indicates the best performance, and blue indicates the second-best.
Table 5. The comparison of average PSNR gain over the UVG and MCL-JCV datasets. Red indicates the best performance, and blue indicates the second-best.
DatasetQPRR-DnCNN v2FM-VSREDVRBasicVSR++EOCVSR
UVG3235.35235.35835.45035.52635.558
3733.14233.16733.15133.23733.250
4230.68330.70130.75530.82230.822
4728.26728.24328.32628.33128.375
Overall31.86131.86731.92031.97932.001
MCL-JCV3234.86034.93535.05335.14035.172
3732.74832.74932.77432.89032.930
4230.42630.48030.50030.57530.568
4728.22228.30428.31328.32828.386
Overall31.56431.61731.66031.73331.764
Table 6. The comparison of BD-BR over HEVC standard test sequences. Red indicates the best performance, and blue indicates the second-best.
Table 6. The comparison of BD-BR over HEVC standard test sequences. Red indicates the best performance, and blue indicates the second-best.
DatasetClassSequencesRR-DnCNN v2FM-VSREDVRBasicVSR++EOCVSR
HEVCAPeopleOnStreet−11.621−13.780−15.541−18.113−20.621
Traffic−3.916−5.019−6.153−7.833−8.584
BBasketballDrive0.687−7.133−11.365−14.361−14.140
Cactus−0.224−9.965−12.462−14.842−15.386
Kimono−13.743−17.421−20.907−22.182−22.501
ParkScene5.323−4.408−7.121−8.035−8.452
CBasketballDrill0.307−3.147−4.883−6.685−8.300
RaceHorses−6.599−7.366−8.185−10.610−11.420
EFourPeople−11.528−12.154−12.788−13.591−14.836
Johnny−19.018−19.580−20.076−21.614−22.046
KristenAndSara−7.181−7.877−8.414−10.864−11.721
Average−6.137−9.804−11.627−13.521−14.364
UVGAverage−27.086−27.313−27.804−29.502−29.849
MCL-JCVAverage−24.066−24.526−24.913−27.599−28.217
Table 7. The performance of the EORB with edge-perceiving filters vs. without edge-perceiving filters.
Table 7. The performance of the EORB with edge-perceiving filters vs. without edge-perceiving filters.
EORB w/o Edge-Perceiving FiltersEORB w/ Edge-Perceiving Filters
PSNR (dB)28.31728.386
Table 8. The adaption of the proposed design on video super-resolution denoising task.
Table 8. The adaption of the proposed design on video super-resolution denoising task.
Noise LevelScaleMIRNetEOCVSR
σ = 15 × 2 34.6535.10
σ = 25 × 2 33.8634.45
σ = 50 × 2 31.0632.46
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

Share and Cite

MDPI and ACS Style

Wang, Z.; Quan, G.; He, G. Edge-Oriented Compressed Video Super-Resolution. Sensors 2024, 24, 170. https://doi.org/10.3390/s24010170

AMA Style

Wang Z, Quan G, He G. Edge-Oriented Compressed Video Super-Resolution. Sensors. 2024; 24(1):170. https://doi.org/10.3390/s24010170

Chicago/Turabian Style

Wang, Zheng, Guancheng Quan, and Gang He. 2024. "Edge-Oriented Compressed Video Super-Resolution" Sensors 24, no. 1: 170. https://doi.org/10.3390/s24010170

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop