基于WGAN-GP和U-net改进的图像增强的方法、装置及存储介质

摘要

本发明公开了基于WGAN‑GP和U‑net改进的图像增强的方法、装置及存储介质，其方法包括如下步骤：解封装输入的视频流或文件，得到第一视频码流和第一音频码流；解码所述第一视频码流，得到第一YUV数据；构建WGAN‑GP网络；将所述第一YUV数据输入所述WGAN‑GP网络，经所述WGAN‑GP网络训练输出第二YUV数据；对所述第二YUV数据编码得到第二视频码流；将所述第二视频码流和第一音频码流封装为视频流或文件。本发明通过WGAN‑GP网络训练视频码流，可以提升视频的画质，降低重新采集视频的成本。