qubvel-org / segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

https://smp.readthedocs.io/

qubvel-org/segmentation_models.pytorch Issues

Add `multiclass` fine-tuning example
Updated 10 days ago7
Guidance required to add CBAM Module
Closed 15 days ago1
How to implement model parallelism using PyTorch on an HPC environment?
Updated 17 days ago16
Can't load saved model after training
Updated 17 days ago3
Is this library still maintained?
Updated 22 days ago18
DiceLoss when multiclass mode throwing an assertion error
Updated 23 days ago6
Using multi GPUs
Updated a month ago1
Add multi-gpu training example script
Updated a month ago
Add segmentation + classification head fine-tuning example
Updated a month ago
add UperNet model
Updated a month ago4
add other new models
Updated a month ago3
Segmentation fault (core dumped)
Closed a month ago3
how to calculate the number of parameters and computational complexity for each model
Updated 2 months ago1
How can I get the final evaluation metrics
Updated 2 months ago1
Dice Loss resulting in unexpected logit outputs
Updated 2 months ago2
Understanding difference between TensorFlow and PyTorch implementations of `Unet`
Closed 7 months ago4
multiple object task
Updated 2 months ago1
'DeepLabV3Plus' object has no attribute 'save_pretrained'
Updated 2 months ago12
data preproccessing : mask = to_tensor
Closed 2 months ago5
SSL: CERTIFICATE_VERIFY_FAILED error
Closed 3 months ago1
ModuleNotFoundError: No module named 'segmentation_models_pytorch.unetplusplus'
Updated 3 months ago6
I can't run "cars segmentation (camvid).ipynb" example
Updated 3 months ago2
No problems with all losses, except JaccardLoss
Updated 3 months ago8
the method of adding new networks
Updated 3 months ago4
How to specify class weights for loss function?
Updated 3 months ago1
Adding FeatUp
Updated 4 months ago1
get_preprocessing_fn
Closed 4 months ago2
Extracting features from the model
Closed 4 months ago2
More classes result in smaller model?
Closed 4 months ago2
epsilon in jaccardscore and dicescore is not read as a torch tensor
Closed 4 months ago3
How to correct contribute example with pt-lightning from my fork to original repo
Closed 4 months ago3
examples (notebooks) have not been updated
Closed 4 months ago2
Get score per class
Updated 4 months ago1
tracing mit-b1 on cpu and then running on gpu fails
Closed 4 months ago2
Multiclass segmentation, some basic questions
Closed 5 months ago9
Error in loss calculation when using ignore_index
Closed 5 months ago2
Using "segmentation_models_pytorch" with Unet++
Closed 5 months ago2
Class weights specification on IoU metric
Closed 6 months ago2
UnetDecoder upsampling
Closed 6 months ago2
Training interruption
Closed 6 months ago2
SSL certificate expired for se_resnext50_32x4d encoder
Closed 7 months ago3
Clarify Error Message for Condition Checking 'threshold' in metrics/functional.py
Closed 7 months ago2
modify the code for instance segmentation
Closed 7 months ago2
Network architecture issue
Closed 7 months ago4
Segmentation fault (core dumped) Probelm
Closed 7 months ago2
Suggestion
Closed 8 months ago2
How to use reparameterization with mobileone?
Closed 8 months ago2
'tuple' object has no attribute 'to'
Closed 8 months ago2
TypeError: preprocess_input() missing 1 required positional argument: 'x'
Closed 8 months ago2
UNET is not translation invariant
Closed 10 months ago1