search

xuebinqin / U-2-Net

The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."
Apache License 2.0
8.31k stars 1.43k forks source link

The results are not good when the object has small holes. #379

Open Wananrd opened 5 months ago

Wananrd commented 5 months ago

Recently, I've been working on tasks that involve saliency object detection, and I'm just starting to learn about this field.

I'm using your model, and it doesn't perform well in detecting objects with holes in them, especially in parts. Do you have any suggestions or improvements for this issue?

屏幕截图 2024-02-03 164249

locusbear commented 3 months ago

You can search for trimap, which is the effect of creating an unknown region by erosion dilation, and specifically doing losses on the unknown region to constrain the edge. I remember someone mentioned and implemented this in the issue, search for it