segmentation_models.pytorch

segmentation_models.pytorch copied to clipboard

I can't run "cars segmentation (camvid).ipynb" example

2

Hi. As I understand the below code in **cars segmentation (camvid).ipynb** should be changed. Because I can't run it. Am I right? Do you plan to update the examples? 1....

atekgul

'DeepLabV3Plus' object has no attribute 'save_pretrained'

12

Почему тут есть возможность сохранения модели https://smp.readthedocs.io/en/latest/save_load.html как в документации. А я не могу сохранить модель DeepLabV3Plus? > Why is it possible to save the model here https://smp.readthedocs.io/en/latest/save_load.html as in...

Marcus2331

No problems with all losses, except JaccardLoss

8

Hi ! A quick question about a loss, I've used quite a few losses and loss mixtures without any problems, but when I use iou loss, I only get empty...

JonasZaoui

the method of adding new networks

4

How can i introduce my network in this frame, and i can adjust this net later

wangluoWSG

How to specify class weights for loss function?

1

How to set class weights for different loss functions? Like Jaccard Loss

tanmay4269

DiceLoss when multiclass mode throwing an assertion error

10

my circumstance is as follows and they are just examplary code: ```python y_pred.size() : (5, 4, 256, 384) # B, C, H, W` y_true.size() : (5, 4, 256, 384) #...

Foundsheep

Bumps [timm](https://github.com/huggingface/pytorch-image-models) from 0.9.7 to 1.0.9. Release notes Sourced from timm's releases. Release v1.0.9 Aug 21, 2024 Updated SBB ViT models trained on ImageNet-12k and fine-tuned on ImageNet-1k, challenging quite...

dependabot[bot]

Add `multiclass` fine-tuning example

2

It would be great to have more examples for fine-tuning in the library! Similar to current examples for [binary-segmentaion](https://github.com/qubvel-org/segmentation_models.pytorch/blob/main/examples/binary_segmentation_intro.ipynb) and [multi-label](https://github.com/qubvel-org/segmentation_models.pytorch/blob/main/examples/cars%20segmentation%20(camvid).ipynb) segmentation would be great to have multi-class segmentation example...

qubvel

Using multi GPUs

1

Hi, I wrapped models in DataParallel but it does not use multiGPUs to train. Is there any lead for this? Thank you!

tuvovan

How to implement model parallelism using PyTorch on an HPC environment?

16

Hello, I am trying to implement model parallelism using PyTorch on my HPC environment, which has 4 GPUs available. My goal is to split a neural network model across these...

Akshara211