Max Pooling with Coordinates

Medium · Solved in NumPy · Machine learning coding practice

Problem

Implement a 2D max pooling operation that returns the coordinates of the maximum values, enabling unpooling operations used in architectures like SegNet for semantic segmentation.

Topics: cnn, computer-vision, pooling

The full statement, worked examples, hints and the test suite are available once you sign in. You can then solve Max Pooling with Coordinates in the browser and run it against the tests.

Related problems

• Max Pooling Easy
• Optimizers: From Momentum to Adam Medium
• AdamW Optimizer Medium
• Implement Bagging Classifier with Decision Stumps Medium
• Basic KV Cache: Prefill and Decode Medium
• Binary Vector Scoring with Lookup Tables Medium