Hands-On Java Deep Learning for Computer Vision

Pooling on RGB images

So far, we've seen the pooling layer operations, max and average pooling, with a two-dimensional matrix. But, as we saw, RGB images and color images are three-dimensional matrices, with three channels, so the value of the third dimension is three:

Similar to the convolution, we'll also take three filters, and for each of the filters we'll do the operation we saw in the previous section independently. The filter will handle the pooling layer operation with the channel. It will, of course, produce three two-dimensional matrices, which will be the product of the three independent operations.

Now, if you recall from the convolution, ...