segmentation_models.pytorch
detectron2
segmentation_models.pytorch | detectron2 | |
---|---|---|
14 | 49 | |
8,844 | 28,783 | |
- | 1.4% | |
4.1 | 7.6 | |
8 days ago | 5 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
detectron2
-
Ask HN: How to train an image recognition AI
I don’t do AI professionally but as a hobby, so this may not be the best way. But the way you described, it seems the user maybe taking the picture a bit further away and there may be other objects in the frame. So you may want to look into some sort of segmentation or have bounding box. This could help the user make sure they are looking at documents for the correct machine.
I think something like detectron2 [1] could help. It is Apache2 license, so commercial friendly. That said the pre-trained weights may not be used for commercial purposes, so you’ll want to check on that.
[1] https://github.com/facebookresearch/detectron2
-
Instance segmentation of small objects in grainy drone imagery
And not enough true positives either. Add more augmentations in the config. Also make sure the config is set correctly, so that Detectron2 isn't skipping background images: https://github.com/facebookresearch/detectron2/issues/80
- Openpose alternatives (humanSD & Densepose)
-
Probelms with importing tensormask from detectron2.projects
I followed the setup of https://github.com/facebookresearch/detectron2/tree/main/projects/TensorMask. But still I can not import it. As I can with from detectron2.projects import point_rend easily from PointRend projects
-
Problems with Lazy Config detectron2 (MViTv2)
I have to use this config file with the dataloader which is in https://github.com/facebookresearch/detectron2/blob/main/projects/MViTv2/configs/common/coco_loader.py. I figured that i can use cfg.dataloader.train.dataset.names = "my_dataset_train" for this.
-
"[D]" Problems with Lazy Config detectron2 (MViTv2)
I want to use this config file https://github.com/facebookresearch/detectron2/blob/main/projects/MViTv2/configs/mask_rcnn_mvitv2_t_3x.py like the beneath typical way I use a yaml config file. But giving so many errors one after another that, I even failed to count at this point.
-
AI Real Time (lgd for cn)
Which is built on https://github.com/facebookresearch/detectron2
-
List of AI-Models
Click to Learn more...
-
good computer vision or deep learning projects in github
Detectron2 (GitHub: https://github.com/facebookresearch/detectron2) is a Facebook AI Research library with state-of-the-art object detection and segmentation algorithms in PyTorch.
- Object Detection using PyTorch: Would you recommend a Framework (Detectron2, MMDetection, ...) or a project from scratch ?
What are some alternatives?
yolact - A simple, fully convolutional model for real-time instance segmentation.
mmdetection - OpenMMLab Detection Toolbox and Benchmark
mmsegmentation - OpenMMLab Semantic Segmentation Toolbox and Benchmark.
yolov5 - YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
face-parsing.PyTorch - Using modified BiSeNet for face parsing in PyTorch
openpose - OpenPose: Real-time multi-person keypoint detection library for body, face, hands, and foot estimation
EfficientNet-PyTorch - A PyTorch implementation of EfficientNet and EfficientNetV2 (coming soon!)
U-2-Net - The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."
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.
pytorch-lightning - Build high-performance AI models with PyTorch Lightning (organized PyTorch). Deploy models with Lightning Apps (organized Python to build end-to-end ML systems). [Moved to: https://github.com/Lightning-AI/lightning]
pyannote-audio - Neural building blocks for speaker diarization: speech activity detection, speaker change detection, overlapped speech detection, speaker embedding
rembg - Rembg is a tool to remove images background