TensorFlow-Advanced-Segmentation-Models icon indicating copy to clipboard operation
TensorFlow-Advanced-Segmentation-Models copied to clipboard

Published 20 hours ago • verified publisher icon JanMarcelKezmann

PAM and CAM modules are hardcoded for channels_first

Open tabmoo opened this issue 3 years ago • 0 comments

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?

tabmoo avatar Apr 14 '22 10:04 tabmoo