fch1993 写了:mawen1250 写了:NAVras 写了:
啊嘞,我以为caffe的cudnn速度是最快的
英伟达这个cudnn真是白优化了(噗)
cuDNN快得多,但有些人可能系统环境方面不知道哪里出了问题,导致速度没有提升。另外就是caffe版和w2xc的最优block设置很不一样。
我不知道你是什么显卡出来的情况
我自己测试以各自block的最快速情况来说。
GTX670公版
w2xc r6
caffe r2
N卡驱动最新版
NVIDIA GPU Computing Toolkit v7.5
CuDNN v4
测试速度480p到720p 两者差距大概caffe快约0.01fps不到
res = mvf.Depth(res,depth=32)
#res = core.w2xc.Waifu2x(res,noise=0, scale=2, block=512, photo=False, gpu=1)
res = core.caffe.Waifu2x(res,noise=0, scale=2, block=256, photo=False, cudnn=True, tta=False)
res.set_output()
r9版本用upconv_7_anime_style_art_rgb这个model的话N卡速度暴增,我怀疑旗舰显卡都能满足实时播放的要求了。不过现在实时播放用madVR的NGU算法效果也很赞。