search

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
159 stars 19 forks source link

how to visualize the enhanced yuv data #22

Closed doudoublues closed 1 year ago

doudoublues commented 1 year ago

thanks for your work!, After inference, the y-channel has been enhanced, I want to know how to visualize the enhanced y-channel with origin U,V data by transforming them to RGB data?

ryanxingql commented 1 year ago

Hi, You may use this code: https://github.com/ryanxingql/pythonutils/blob/09f6795c2d686bf1f4e9ce904ff9e134438ceab7/conversion.py#L172 to convert YUV420P to YUV444P.

Then convert this YUV444P image to RGB.

doudoublues commented 1 year ago

thanks!