Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/kautenja/keras-pyramid-pooling-module

An implementation of the Pyramid Pooling Module as a Keras layer.
https://github.com/kautenja/keras-pyramid-pooling-module

keras pyramid-pooling-module

Last synced: 25 days ago
JSON representation

An implementation of the Pyramid Pooling Module as a Keras layer.

Awesome Lists containing this project

README

        

# Keras Pyramid Pooling Module

[![PackageVersion][pypi-version]][pypi-home]
[![PythonVersion][python-version]][python-home]
[![Stable][pypi-status]][pypi-home]
[![Format][pypi-format]][pypi-home]
[![License][pypi-license]](LICENSE)

[pypi-version]: https://badge.fury.io/py/keras-pyramid-pooling-module.svg
[pypi-license]: https://img.shields.io/pypi/l/keras-pyramid-pooling-module.svg
[pypi-status]: https://img.shields.io/pypi/status/keras-pyramid-pooling-module.svg
[pypi-format]: https://img.shields.io/pypi/format/keras-pyramid-pooling-module.svg
[pypi-home]: https://badge.fury.io/py/keras-pyramid-pooling-module
[python-version]: https://img.shields.io/pypi/pyversions/keras-pyramid-pooling-module.svg
[python-home]: https://python.org

**Deprecated:** Please use [tfa.layers.SpatialPyramidPooling2D](https://www.tensorflow.org/addons/api_docs/python/tfa/layers/SpatialPyramidPooling2D) instead.

A [Keras](https://keras.io) implementation of the Pyramid Pooling Module
discussed in [_Pyramid scene parsing network [1]_](#references).

![layer](https://user-images.githubusercontent.com/2184469/109330138-f9ca3f80-7820-11eb-9fbb-4a3b89a88bea.png)

## Installation

The preferred installation of `keras-pyramid-pooling-module` is from `pip`:

```shell
pip install keras-pyramid-pooling-module
```

## Usage

The module functions as any other convolutional / pooling layer applied to a
rank 4 tensor (batch, height, width, channels):

```python
from keras.layers import Input
from keras.models import Model
from keras_pyramid_pooling_module import PyramidPoolingModule

input_ = Input((224, 224, 3))
x = PyramidPoolingModule()(input_)
model = Model(inputs=input_, outputs=x)
```

See [PyramidPoolingModule.ipynb](PyramidPoolingModule.ipynb) for a rough example
of the layer outputs for a simple RGB image.

See [CIFAR10Classifier.ipynb](CIFAR10Classifier.ipynb) for an example of using
the layer in a model to classify images.

## References

[_[1] H. Zhao, J. Shi, X. Qi, X. Wang, and J. Jia. Pyramid scene parsing network. CVPR, 2017._](https://hszhao.github.io/projects/pspnet/)