search

qubvel-org / segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
https://smp.readthedocs.io/
MIT License
9.35k stars 1.64k forks source link

multiple object task #897

Open water367 opened 1 month ago

water367 commented 1 month ago

hello, thanks for your contribution to the segmentation community!

I am designing a multi-objective task based on smp but failed. Does the framework support custom objective functions and training of multi-output models?

qubvel commented 1 month ago

Hi @water367, can you provide a bit more details on that?