Batch-aware loss function? about lcfcn HOT 3 OPEN

Hiya @yassineAlouini , what you suggested should work! It might be a bit slow, and the learning rate might need to be adjusted.

LCFCN does not work natively with batch sizes larger than one, mainly because the watershed algorithm does not support more than one image. If there is a Pytorch-based watershed method, then we could incorporate it and make LCFCN support larger batches.

Thanks for the nice, concise code in making the code support batches larger than 1.

from lcfcn.

Indeed, watershed is probably the limiting part for now. I will have a look at a pytorch implementation and update with more details. Thanks for the quick feedback.

from lcfcn.

I have found this: https://paperswithcode.com/paper/deep-watershed-transform-for-instance. Not sure how useful it could be, I probably need some exploration to see if we can extract the watershed part of it.

Here is a Pytorch implementation: https://github.com/timothyn617/watershed-transform

from lcfcn.