FLOPs of PWC-Net

Open cjsoft opened this issue 5 years ago • 1 comments

There is cpp extension correlation_cuda in PWC-Net. thop library won't be able to calculate a custom extension.

I'm doing a benchmark research on some dl based flow net. It will be very nice of you, if you can provide formula to calculate PWC-net FLOPs count for arbitrary image size.

Best regards.

Mar 24 '20 09:03 cjsoft

You could hand on the calculation, like what I did here: It's basically matrix multiplication.

Jun 05 '20 08:06 ProNoobLi