Implementing RoI Pooling in TensorFlow Keras