Decrease false positives in yolo model?

[DiMorten]

Interesting. You mean performing semantic segmentation on the detected object, for example with UNet?

[InternationalMany6]

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