Pavel Iakubovskii

icon company Vision MLE 🤗 @huggingface

Results 253 comments of Pavel Iakubovskii

Output n_class is less than len(class_name) in training?

Your model does not predict some classes, don't see anything strange here suppose you have classes: car, road, sky, grass but on the picture you have only: road, grass and...

preprocessing fn not working

preprocessing function expect image with shape H, W, C

Cannot import trained model

Yes, downgrade to previous version please to load the model. You can save weights only, and it will work across different versions

Module not found

Utils is deprecated, but in case you need it: `from segmentation_models_pytorch import utils`

Module not found

Pytorch lightning âš¡ New example https://github.com/qubvel/segmentation_models.pytorch/blob/master/examples/binary_segmentation_intro.ipynb

Colab notebook not found

[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/qubvel/segmentation_models.pytorch/blob/master/examples/binary_segmentation_intro.ipynb) Check this one, please P.S. Updated

Locked version of `timm` dependency

The reason is that API tends to change, especially in such quick-developing libraries as timm! In one day you can appear with not working code because of that, I prefer...

Locked version of `timm` dependency

I have updated timm version, but a new version, unfortunately, breaks the code https://github.com/qubvel/segmentation_models.pytorch/runs/7586507780?check_suite_focus=true

Locked version of `timm` dependency

I will fix and update it later, there are some changes, not seeing a quick fix for it

Best way to integrate with fastai?

Hi @jph00 Thanks for connecting! I am very sorry for not answering before, the project has not been maintained for a while and I had some other vital issues. I...