ryanxingql / stdf-pytorch

Implementation of "Spatio-Temporal Deformable Convolution for Compressed Video Quality Enhancement" (AAAI'20).
https://www.aiide.org/ojs/index.php/AAAI/article/view/6697
Apache License 2.0
155 stars 20 forks source link

How to save the enhanced video? #9

Closed linjing7 closed 3 years ago

linjing7 commented 3 years ago

Hi, the model output is Y channel, could you please tell me how to save it as an enhanced result shown in the paper?

ryanxingql commented 3 years ago

You can combine the enhanced Y channel with the original U、V channels, such that the enhanced YUV frame can be generated.

linjing7 commented 3 years ago

Okay, thank you very much.