segmentation

There are 66 repositories under segmentation topic.

WZMIAOMIAO / deep-learning-for-image-processing
deep learning for image processing including classification and object-detection etc.
bilibili classification deep-learning object-detection pytorch segmentation tensorflow2
Language:Python 20802
albumentations
albumentations-team / albumentations
Fast image augmentation library and an easy-to-use wrapper around other libraries. Documentation: https://albumentations.ai/docs/ Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
image-augmentation machine-learning augmentation deep-learning detection fast-augmentations segmentation image-segmentation image-processing image-classification python object-detection
Language:Python 13412
lexfridman / mit-deep-learning
Tutorials, assignments, and competitions for MIT Deep Learning related courses.
deep-learning mit self-driving-cars deep-reinforcement-learning deep-rl artificial-intelligence neural-networks segmentation machine-learning data-science deeplearning tensorflow tensorflow-tutorials jupyter-notebooks
Language:Jupyter Notebook 10000
qubvel / segmentation_models.pytorch
Segmentation models with pretrained backbones. PyTorch.
deeplab-v3-plus deeplabv3 fpn hacktoberfest image-processing image-segmentation imagenet linknet models pretrained-backbones pretrained-models pretrained-weights pspnet pytorch segmentation segmentation-models semantic-segmentation unet unet-pytorch unetplusplus
Language:Python 8797
utkuozbulak / pytorch-cnn-visualizations
Pytorch implementation of convolutional neural network visualization techniques
cam cnn-visualization deep-dream grad-cam gradient gradient-visualization guided-backpropagation guided-grad-cam pytorch saliency segmentation smooth-grad
Language:Python 7707
jetson-inference
dusty-nv / jetson-inference
Hello AI World guide to deploying deep-learning inference networks and deep vision primitives with TensorRT and NVIDIA Jetson.
caffe computer-vision deep-learning digits embedded image-recognition inference jetson jetson-nano jetson-tx1 jetson-tx2 jetson-xavier jetson-xavier-nx machine-learning nvidia object-detection robotics segmentation tensorrt video-analytics
Language:C++ 7330
mikel-brostrom / yolo_tracking
BoxMOT: pluggable SOTA tracking modules for segmentation, object detection and pose estimation models
strongsort bytetrack ocsort osnet deep-learning segmentation tensorrt tracking-by-detection yolo botsort deepocsort multi-object-tracking mot mots multi-object-tracking-segmentation
Language:Python 6093
MIC-DKFZ / nnUNet
segmentation
Language:Python 5026
qubvel / segmentation_models
Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
unet fpn segmentation keras pretrained pre-trained image-segmentation linknet pspnet tensorflow segmentation-models resnet resnext efficientnet densenet keras-tensorflow keras-models tensorflow-keras keras-examples mobilenet
Language:Python 4603
PaddlePaddle / PaddleX
PaddlePaddle End-to-End Development Toolkit（『飞桨』深度学习全流程开发工具）
classification deep-neural-networks deeplearning deployment detection industry jetson mobilenet neural-networks resnet segmentation unet yolov3
Language:Python 4566
charlesq34 / pointnet
PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
classification geometry-processing neural-network point-cloud segmentation tensorflow
Language:Python 4564
zhixuhao / unet
unet for image segmentation
unet keras segmentation
Language:Jupyter Notebook 4433
DefTruth / lite.ai.toolkit
🛠 A lite C++ toolkit of awesome AI models, support ONNXRuntime, MNN. Contains YOLOv5, YOLOv6, YOLOX, YOLOR, FaceDet, HeadSeg, HeadPose, Matting etc. Engine: ONNXRuntime, MNN.
yolox retinaface onnxruntime segmentation yolor yolop nanodet robustvideomatting mnn ncnn tnn modnet scrfd yolox-nano yolox-tiny yolov6 yolov7 onnx yolov5 yolov8
Language:C++ 3434
SysCV / sam-hq
Segment Anything in High Quality [NeurIPS 2023]
sam segmentation segment-anything zero-shot-segmentation high-quality segment-anything-model
Language:Python 3351
yanx27 / Pointnet_Pointnet2_pytorch
PointNet and PointNet++ implemented by pytorch (pure python) and on ModelNet, ShapeNet and S3DIS.
pointnet pointnet2 pointcloud segmentation s3dis modelnet shapenet pytorch classification visualization point-cloud
Language:Python 3187
CharlesShang / FastMaskRCNN
Mask RCNN in TensorFlow
coco detection fasterrcnn maskrcnn pfn rcnn segmentation tensorflow
Language:Python 3098
HRNet / HRNet-Semantic-Segmentation
The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semantic segmentation for HRNet. https://arxiv.org/abs/1908.07919
segmentation semantic-segmentation cityscapes pascal-context lip high-resolution high-resolution-net hrnets transformer segmentation-transformer
Language:Python 3049
charlesq34 / pointnet2
PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Space
point-cloud deep-learning classification segmentation 3d-shape
Language:Python 2941
osmr / imgclsmob
Sandbox for training deep learning networks
3d-face-reconstruction chainer cifar classification deep-learning gluon human-pose-estimation image-classification imagenet keras machine-learning mxnet neural-network pretrained-models pytorch segmentation semantic-segmentation tensorflow tensorflow2
Language:Python 2917
PaddlePaddle / awesome-DeepLearning
深度学习入门课、资深课、特色课、学术案例、产业实践案例、深度学习知识百科及面试题库The course, case and knowledge of Deep Learning and AI
cnn rnn detection classification gan video segmentation pose nlp recommender-system reinforcement-learning bert tinybert dynabert transformer dssm sarsa dqn
Language:Jupyter Notebook 2822
roytseng-tw / Detectron.pytorch
A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
mask-rcnn pytorch detection pose-estimation segmentation detectron
Language:Python 2816
opengeos / segment-geospatial
A Python package for segmenting geospatial data with the Segment Anything Model (SAM)
artificial-intelligence deep-learning geopython geospatial machine-learning segment-anything segmentation
Language:Python 2654
BlenderProc
DLR-RM / BlenderProc
A procedural Blender pipeline for photorealistic training image generation
3d-engines 3d-front-dataset 3d-graphics 3d-reconstruction blender blender-installation blender-pipeline camera-positions camera-sampling computer-graphics depth-images pose-estimation python rendering segmentation suncg-scene synthetic synthetic-data
Language:Python 2542
GeorgeSeif / Semantic-Segmentation-Suite
Semantic Segmentation Suite in TensorFlow. Implement, train, and test new Semantic Segmentation models easily!
computer-vision dataset deep-learning densenet encoder-decoder epoch iou python refinenet segmentation semantic-segmentation semantic-segmentation-models tensorflow upsampling
Language:Python 2492
ShawnBIT / UNet-family
Paper and implementation of UNet-related model.
pytorch segmentation unet
Language:Python 2423
torch-points3d
torch-points3d / torch-points3d
Pytorch framework for doing deep learning on point clouds.
deep-learning kpconv minkowskiengine point-cloud pointnet pytorch s3dis scannet segmentation
Language:Python 2396
abhineet123 / Deep-Learning-for-Tracking-and-Detection
Collection of papers, datasets, code and other resources for object tracking and detection using deep learning
code-collection deep-learning detection object-detection optical-flow paper-collection papers segmentation tracking tracking-by-detection
Language:HTML 2380
xinario / awesome-gan-for-medical-imaging
Awesome GAN for Medical Imaging
generative-adversarial-network medical-imaging deeplearning ct-denoising segmentation medical-image-synthesis detection reconstruction gan registration super-resolution
2230
MrGiovanni / UNetPlusPlus
[IEEE TMI] Official Implementation for UNet++
medical-imaging segmentation unet
Language:Python 2167
rsennrich / subword-nmt
Unsupervised Word Segmentation for Neural Machine Translation and Text Generation
neural-machine-translation segmentation machine-translation nmt subword-units bpe
Language:Python 2153
zhulf0804 / 3D-PointCloud
Papers and Datasets about Point Cloud.
autonomous-driving classification completion datasets detection generation monocular papers point-cloud registration segmentation
Language:Python 2114
mapbox / robosat
Semantic segmentation on aerial and satellite imagery. Extracts features such as: buildings, parking lots, roads, water, clouds
satellite-imagery aerial-imagery machine-learning segmentation openstreetmap
Language:Python 1996
IDEA-Research / detrex
detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.
detr object-detection pytorch dino state-of-the-art dab-detr deformable-detr conditional-detr dn-detr group-detr h-detr mask-dino anchor-detr deta pose-estimation segmentation
Language:Python 1818
bigmb / Unet-Segmentation-Pytorch-Nest-of-Unets
Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet
unet pytorch imagesegmentation segmentation tensorvision torchvision torch python3
Language:Python 1744
VainF / DeepLabV3Plus-Pytorch
Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes
cityscapes deeplabv3 deeplabv3plus pascal-voc pytorch segmentation
Language:Python 1708
apple / ml-cvnets
CVNets: A library for training computer vision networks
ade20k classification computer-vision deep-learning detection imagenet machine-learning mscoco pascal-voc pytorch segmentation
Language:Python 1670

segmentation

WZMIAOMIAO / deep-learning-for-image-processing

albumentations-team / albumentations

lexfridman / mit-deep-learning

qubvel / segmentation_models.pytorch

utkuozbulak / pytorch-cnn-visualizations

dusty-nv / jetson-inference

mikel-brostrom / yolo_tracking

MIC-DKFZ / nnUNet

qubvel / segmentation_models

PaddlePaddle / PaddleX

charlesq34 / pointnet

zhixuhao / unet

DefTruth / lite.ai.toolkit

SysCV / sam-hq

yanx27 / Pointnet_Pointnet2_pytorch

CharlesShang / FastMaskRCNN

HRNet / HRNet-Semantic-Segmentation

charlesq34 / pointnet2

osmr / imgclsmob

PaddlePaddle / awesome-DeepLearning

roytseng-tw / Detectron.pytorch

opengeos / segment-geospatial

DLR-RM / BlenderProc

GeorgeSeif / Semantic-Segmentation-Suite

ShawnBIT / UNet-family

torch-points3d / torch-points3d

abhineet123 / Deep-Learning-for-Tracking-and-Detection

xinario / awesome-gan-for-medical-imaging

MrGiovanni / UNetPlusPlus

rsennrich / subword-nmt

zhulf0804 / 3D-PointCloud

mapbox / robosat

IDEA-Research / detrex

bigmb / Unet-Segmentation-Pytorch-Nest-of-Unets

VainF / DeepLabV3Plus-Pytorch

apple / ml-cvnets