Would you please provide the avg pooling method?

Maratyszcza / NNPACK

Acceleration package for neural networks on multi-core CPUs

BSD 2-Clause "Simplified" License

1.68k stars 317 forks source link

Would you please provide the avg pooling method? #16

Open abangdd opened 8 years ago

abangdd commented 8 years ago

since global avg pooling is essential to the Inception and ResNet architecture，will you implement avg pooling in the future？ thank you

liuyi999111 commented 7 years ago

In pooling-output.c this file seems to have a bug in line 51. Const size_t t = x * stride_height + j - padding_left; should be: Const size_t t = x * stride_width + j - padding_left;

Maratyszcza commented 7 years ago

@liuyi999111 The bug is fixed in 1a75285dab4b76eb3370b46bf985f3bde44d3964. Thank you for reporting.

liuyi999111 commented 7 years ago

The mean pooling method looks easy on the basis of the max pooling method. I can help you implement these simple functions.