好嘞,今天跟大家唠唠我这几天折腾 TecoGAN 的事儿。一开始我也就是图个新鲜,想看看这玩意儿到底能把模糊视频或者图片增强到啥程度。
得把环境搭起来。我是个习惯用 Anaconda 的人,所以第一步就是新建一个虚拟环境,专门给 TecoGAN 用。省得跟其他项目的依赖搞混了,到时候出问题,查都查不明白。
conda create -n tecogan python=3.8
conda activate tecogan
环境建好之后,就开始装依赖。按照 GitHub 上面的文档,一个个 pip install 就行。不过这里有个坑,就是 TensorFlow 的版本。 TecoGAN 好像对 TensorFlow 版本有要求,我试了好几个版本, 1.15.0 才能跑起来。 千万要注意版本问题!
pip install tensorflow==1.15.0
pip install tensorflow-gpu==1.15.0 # 如果你有 NVIDIA 显卡
pip install -r * # 安装其他依赖
依赖装好之后,就可以开始下载预训练模型了。这步也很关键,因为 TecoGAN 需要预训练好的模型才能进行图像增强。 模型文件挺大的,得耐心等等。
模型下好之后,就开始跑 demo。 GitHub 上面有提供一些示例视频,我随便选了一个跑了一下。 刚开始跑的时候,速度那叫一个慢,我的老电脑吭哧吭哧的,风扇狂转。 还好我有点耐心,等它跑完之后,效果还真不错! 原本模糊的视频,变得清晰多了,细节也更丰富了。虽然比不上高清视频,但是至少能看清里面的内容了。
后来我又试着处理了一些自己的老照片。 以前手机像素低,拍出来的照片都很模糊,用 TecoGAN 处理了一下,效果也挺明显的。感觉就像把老照片“复活”了一样,回忆都变得清晰了。
TecoGAN 也不是万能的。 对于一些特别模糊或者损坏严重的视频或者照片,它的效果就比较有限了。 还有就是, TecoGAN 的处理速度比较慢,如果是处理大量视频,可能需要很长时间。
这回折腾 TecoGAN 的经历还是挺有趣的。 虽然中间遇到了一些坑,但是最终还是成功地跑起来了,并且得到了不错的效果。 如果你也有类似的需求,不妨也试试 TecoGAN ,说不定能给你带来惊喜。
总结一下我踩过的坑:
TensorFlow 版本一定要对!
下载预训练模型!
耐心等待!
希望我的经验能对你有所帮助!
补充说明
对了,补充一下,想要跑得快,最好还是用 NVIDIA 的显卡,并且安装 tensorflow-gpu 版本。 我用 CPU 跑的时候,简直慢到怀疑人生。 用 GPU 跑的话,速度能提升好几倍。