Visualization on 2D Learnable masks

Example #1

Example #2

Example #3