search

NVlabs / PWC-Net

PWC-Net: CNNs for Optical Flow Using Pyramid, Warping, and Cost Volume, CVPR 2018 (Oral)
Other
1.63k stars 357 forks source link

FLOPs of PWC-Net #105

Open cjsoft opened 4 years ago

cjsoft commented 4 years ago

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 4 years ago

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