ext_portrait_segmentation icon indicating copy to clipboard operation
ext_portrait_segmentation copied to clipboard

training tip

Open freaad opened this issue 5 years ago • 4 comments

Do you have any tip for small object trainning. I am dealing the small binary segmentation but it doesn';t have good result. Do you have any tip.

freaad avatar Jan 19 '20 02:01 freaad

A generic tip is to use focal loss for training with small objects. Not sure if it will work here but it will certainly help in general.

lessw2020 avatar Mar 03 '20 20:03 lessw2020

I have already try but not a good result....

freaad avatar Mar 05 '20 04:03 freaad

I'm testing some other loss functions tomorrow and will let you know if any progress. Boundary loss penalty looks really good.

lessw2020 avatar Mar 05 '20 05:03 lessw2020

Thank you for answering. I agree the boundary loss penalty is good idea. However my problem is quite imblance data. Background is 95 % the target object is only 5 percent in the image. My problem is only binary segmentation but I cannot get the 90% iou score. I usually get the 83%. I try this network but still no luck. I am trying to use the synchronize batch normalization. Wish me good luck.

freaad avatar Mar 05 '20 17:03 freaad