segmentation_models.pytorch
mmsegmentation
segmentation_models.pytorch | mmsegmentation | |
---|---|---|
14 | 7 | |
8,844 | 7,436 | |
- | 2.1% | |
4.1 | 8.2 | |
8 days ago | 12 days ago | |
Python | Python | |
MIT License | Apache License 2.0 |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
segmentation_models.pytorch
-
Instance segmentation of small objects in grainy drone imagery
Also, I’d suggest considering switching to the segmentation-models library - it provides U-Net models with a variety of pretrained backbones of as encoders. The author also put out a PyTorch version. https://github.com/qubvel/segmentation_models.pytorch https://github.com/qubvel/segmentation_models
-
[D] Improvements/alternatives to U-net for medical images segmentation?
SMP offers a wide variety of segmentation models with the option to use pre-trained weights.
-
Improvements/alternatives to U-net for medical images segmentation?
SMP has a lot of different choices for architecture other than unet, and a ton of different encoders. I like deeplabv3+/unet with regnety encoder, works well for most things https://github.com/qubvel/segmentation_models.pytorch
-
Medical Image Segmentation Human Retina
This basic example from segmentation models PyTorch repo would be good tutorial to start with. The library is very good, I like the unet, fpn and deeplabv3+ architectures with regnety as encoder https://github.com/qubvel/segmentation_models.pytorch/blob/master/examples/binary_segmentation_intro.ipynb
-
Automatic generation of image-segmentation mask pairs with StableDiffusion
Sounds like a good semantic segmentation problem, I like this repo: https://github.com/qubvel/segmentation_models.pytorch
-
Dice Score not decreasing when doing semantic segmentation
When i pass the CT-Scans and the masks to the Loss Function, which is the Jaccard-Loss from the segmentation_models.pytorch library, the value does not decrease but stay in the range of 1.0-0.9 over 50 epochs training on only one batch of 32 images. As far as I have understood, my network should overfit and the loss should decrease since I am only training on one batch of a small amount of images. However this does not happen. I also tried more batches with all the data over 100 epochs, but the loss does not decrease either obviously. Does anyone have an idea what I might have done wrong? Do I have to change anything when passing the masks to my loss function?
-
Good Brain Tumor segmentation model !?
I know there is a decent one in segmentation models python (MA-Net: A Multi-Scale Attention Network for Liver and Tumor Segmentation)
-
Advice needed
You could also use qubvel's segmentation models if you would like to explore semantic segmentation.
-
[D][R] Is there a standard architecture for U-Nets, pixel-to-pixel models, VAEs, and the like?
Check out segmentation models pytorch, really easy to use, has a great interface.
-
Pytorch GPU Memory Leak Problem: Cuda Out of Memory Error !!
Have you tried another implementation? For example: qubvel/segmentation_models.pytorch
mmsegmentation
- [D] The MMSegmentation library from OpenMMLab appears to return the wrong results when computing basic image segmentation metrics such as the Jaccard index (IoU - intersection-over-union). It appears to compute recall (sensitivity) instead of IoU, which artificially inflates the performance metrics.
-
Is there any ML model out there for room surfaces detection? (ceiling, floor, windows)
Segmentation models trained on datasets like ADE20k could probably be used for that, because it has separate classes for these things iirc. https://github.com/open-mmlab/mmsegmentation should have suitable pretrained models available.
-
MMDeploy: Deploy All the Algorithms of OpenMMLab
MMSegmentation: OpenMMLab semantic segmentation toolbox and benchmark.
- Mmsegmentation - Openmmlab semantic segmentation toolbox and benchmark.
- Mmsegmentation – Openmmlab semantic segmentation toolbox and benchmark
-
Semantic Segmentation models
This repo is amazing: https://github.com/open-mmlab/mmsegmentation
-
What's A Simple Custom Segmentation Pipeline?
Mmsegmentation would be a good place to start for basic segmentation. They have lots of recent methods and pretained models you could fine-tune from. They also support quite a few datasets including VOC. There is a custom dataset format which looks straightforward to create.
What are some alternatives?
yolact - A simple, fully convolutional model for real-time instance segmentation.
Pytorch-UNet - PyTorch implementation of the U-Net for image semantic segmentation with high quality images
face-parsing.PyTorch - Using modified BiSeNet for face parsing in PyTorch
Swin-Transformer-Semantic-Segmentation - This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation.
EfficientNet-PyTorch - A PyTorch implementation of EfficientNet and EfficientNetV2 (coming soon!)
Mask_RCNN - Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow
SegmentationCpp - A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: VGG, ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.
pyannote-audio - Neural building blocks for speaker diarization: speech activity detection, speaker change detection, overlapped speech detection, speaker embedding
PaddleSeg - Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image Matting, 3D Segmentation, etc.
efficientdet-pytorch - A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
labelme2coco - A lightweight package for converting your labelme annotations into COCO object detection format.