Smoothing custom segmentation edges
Hi Snap community,
I've been working on the creation of a custom nail segmentation Lens. My current model performs consistently but the quality and security of the segmentation edge, without any post processing, is super fuzzy and not defined enough.
To work around this I created a remap operation where I take the values of the segmentation mask and crush values under 0.12 as seen below.

The only issue with this approach is that currently I have jagged edges. I've been trying to figure out a way of "refining" the edges, or "choke" them, getting them closer to the nail, as it is possible with other segmentations inside Lens Studio but I haven't been able to figure out a way. Do any of you has any good ideas of how to achieve this?

Thank you all for your help. If you are a shader pro but want to know about custom segmentations using ML and custom datasets, super happy to exchange knowledge to jump over this hump.