关于视频编解码硬件加速的问题

[tayton42]

#159 我遇到了和该问题一致的情况，在使用v100时，模型推理速度只有2-3帧，speedtest却有60-70帧，请问是以下代码造成这种问题吗

                if output_foreground is not None:
                    writer_fgr.write(fgr[0])
                if output_alpha is not None:
                    writer_pha.write(pha[0])  # pha[0]  [12,1,h,w]
                if output_composition is not None:
                    if output_type == 'video':
                        com = fgr * pha + bgr * (1 - pha)
                    else:
                        fgr = fgr * pha.gt(0)
                        com = torch.cat([fgr, pha], dim=-3)
                    writer_com.write(com[0])

如果不是，拖慢推理速度的代码在哪里，如果是，我应该怎样加速视频编解码？谢谢！

[Vinlic]

VideoWriter是软编码视频效率会低很多，你可以借助支持硬件加速的ffmpeg对帧进行编码 :D