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

PAM and CAM modules are hardcoded for channels_first #23

Open tabmoo opened 2 years ago

tabmoo commented 2 years ago

It seems that the lines with BS, C, H, W =x.shape are good only for channels_first and there is no code for BS, H, W, C.

Also, what are the axes for softmax in those 2 cases?