search

qubvel-org / segmentation_models.pytorch

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

is there any notebook for multiclass segmentation with pytorch lighning #558

Closed johnahjohn closed 2 years ago

github-actions[bot] commented 2 years ago

This issue is stale because it has been open 60 days with no activity. Remove stale label or comment or this will be closed in 7 days.

github-actions[bot] commented 2 years ago

This issue was closed because it has been stalled for 7 days with no activity.

davodogster commented 2 years ago

@qubvel please reopen it :) @johnahjohn

JulienMaille commented 2 years ago

There's an example using pytorch_lightning now: https://github.com/qubvel/segmentation_models.pytorch/blob/master/examples/binary_segmentation_intro.ipynb

davodogster commented 2 years ago

@JulienMaille Thanks but no, that is for binary not multiclass seg. The two are quite different to implement. Please let me know if you have a multiclass example! Cheers, Sam