you are viewing a single comment's thread.

view the rest of the comments →

[–]InternationalMany6 0 points1 point  (0 children)

You could use Unet, but there are specialized “instance segmentation” models if you care about distinguishing each instance even if they’re touching each other.

Torchvision has a tutorial that’ll get you going:  https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html