search

JanMarcelKezmann / TensorFlow-Advanced-Segmentation-Models

A Python Library for High-Level Semantic Segmentation Models based on TensorFlow and Keras with pretrained backbones.
Other
151 stars 46 forks source link

ObjectAttentionBlock2D bug? #15

Closed tabmoo closed 2 years ago

tabmoo commented 2 years ago

Hi!

There is something wrong with self.pool and x. Is one allowed to use self.scale > 1?

JanMarcelKezmann commented 2 years ago

Hi @tabmoo ,

thank you for pointing the error out, I have already pushed a fix for that. The "x" should of course be "feats".