search

pengliuhan / OVQE

[IEEE TBC] PyTorch Implementation for "OVQE: Omniscient Network for Compressed Video Quality Enhancement"
Apache License 2.0
7 stars 2 forks source link

关于内存大小咨询 #1

Open LHL731 opened 1 year ago

LHL731 commented 1 year ago

您好，很喜欢你们的这个工作，我想运行这个代码学习一下，但是我看在测试和训练的时候是将整个序列一起输入网络而不是像STDF方法一类有裁剪的输入网络，我的内存似乎不支持这个输入策略，想咨询下你们运行该代码时的内存策略或者有没有解决办法，期待您的回复

pengliuhan commented 1 year ago

Jensen @.***

（1） 训练的时候您可以将Batchsize调成2或者4，这样的话会增加训练时间，模型收敛后得到的psnr结果应该是基本相同。如果这样还不行的话，那么您可以减小输入视频片段数15->10，这样的话可能会影响psnr。 （2）测试的时候您可以将视频分成若干个小片段，例如一个500帧的视频，拆分成10个50帧的视频输入到网络中测试。但是只这样做的话会影响psnr，您可以采用输入70(10+50+10)帧的形式增强中间50帧，从而得到类似的psnr。 （3）当测试高分辨率(class a，b)视频时。可以将它们分成2快低分辨率视频，这样也可以减小显存的消耗。具体算法可以参考发给您的文件。 （4）结合内存来测试。

------------------ 原始邮件 ------------------ 发件人: "pengliuhan/OVQE" @.>; 发送时间: 2023年3月10日(星期五) 下午2:00 @.>; @.***>; 主题: [pengliuhan/OVQE] 关于内存大小咨询 (Issue #1)

您好，很喜欢你们的这个工作，我想运行这个代码学习一下，但是我看在测试和训练的时候是将整个序列一起输入网络而不是像STDF方法一类有裁剪的输入网络，我的内存似乎不支持这个输入策略，想咨询下你们运行该代码时的内存策略或者有没有解决办法，期待您的回复

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.***>

LHL731 commented 1 year ago

非常感谢您的解答，关于第二个点输入70帧的意思是假如要增强500帧的话，就拆成10个70帧，每个70帧里只算中间50帧的PSNR以弥补拆分带来的PSNR损失吗

pengliuhan commented 1 year ago

是的，这样做相当于弥补了第0帧和最后1帧无法利用相邻帧带来的PSNR损失

Jensen @.***

 

------------------ 原始邮件 ------------------ 发件人: "pengliuhan/OVQE" @.>; 发送时间: 2023年3月10日(星期五) 下午2:46 @.>; @.**@.>; 主题: Re: [pengliuhan/OVQE] 关于内存大小咨询 (Issue #1)

非常感谢您的解答，关于第二个点输入70帧的意思是假如要增强500帧的话，就拆成10个70帧，每个70帧里只算中间50帧的PSNR以弥补拆分带来的PSNR损失吗

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>

LHL731 commented 1 year ago

嗯嗯，非常感谢您的解答，快速地解决了我的问题。