NVlabs/PWC-Net

FLOPs of PWC-Net

cjsoft opened this issue · 1 comments

cjsoft commented

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.

ProNoobLi commented

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