Vybe Guide
Sign Up

Explore Projects

Discover 23 open source projects

Active filters (1):
Search: unetร—
Clear all

Showing 1-20 of 23 projects

Zeyi-Lin/HivisionIDPhotos

AI-powered ID photo generation with face detection and matting

20.8K
Experimental
Python
Computer Vision
General Utilities
FastAPI
#id-photo-generator#face-recognition#image-matting

qubvel-org/segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

11.4K
Stable
Python
Computer Vision
PyTorch
#image-segmentation#semantic-segmentation#pretrained-weights

milesial/Pytorch-UNet

A PyTorch implementation of the U-Net for image semantic segmentation with high quality images.

11.2K
Archived
Python
Computer Vision
PyTorch
#computer-vision#semantic-segmentation#deep-learning

qubvel/segmentation_models

Segmentation models with pretrained backbones for image processing and computer vision tasks.

4.9K
Archived
Python
Computer Vision
Machine Learning Ops
Keras
#image-segmentation#pre-trained#tensorflow

zhixuhao/unet

A Jupyter Notebook implementation of the U-Net convolutional neural network for image segmentation tasks.

4.9K
Archived
Jupyter Notebook
Computer Vision
API Frameworks
Keras
#image-segmentation#deep-learning#computer-vision

MrGiovanni/UNetPlusPlus

Official implementation of the UNet++ medical image segmentation model, useful for computer vision projects.

2.6K
Stable
Python
Computer Vision
API Frameworks
Python
#medical-imaging#segmentation#unet

ShawnBIT/UNet-family

A PyTorch implementation of the popular UNet neural network architecture for image segmentation tasks.

2.6K
Archived
Python
Computer Vision
API Frameworks
PyTorch
#unet#image-segmentation#computer-vision

wolny/pytorch-3dunet

Efficient 3D U-Net model for volumetric semantic segmentation of medical images, built with PyTorch.

2.4K
Stable
Jupyter Notebook
Computer Vision
API Frameworks
PyTorch
#3d-segmentation#semantic-segmentation#volumetric-data

HuCaoFighting/Swin-Unet

Swin-Unet is a pure transformer-based model for medical image segmentation, with potential use in AI-powered coding tools.

2.3K
Experimental
Python
Computer Vision
AI App Builders
Python
#medical-imaging#image-segmentation#transformer-model

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

A PyTorch implementation of various Unet models for image segmentation, including Attention Unet and Nested Unet.

2.2K
Archived
Python
Computer Vision
API Frameworks
PyTorch
#image-segmentation#unet#computer-vision

zoubohao/DenoisingDiffusionProbabilityModel-ddpm-

A simple implementation of the Denoising Diffusion Probability Model (DDPM) for training a U-Net on the CIFAR-10 dataset.

2.2K
Archived
Python
ML Ops
Computer Vision
Python
#ddpm#denoising#computer-vision

jakeret/tf_unet

A generic U-Net implementation in TensorFlow for image segmentation tasks.

1.9K
Archived
Python
Computer Vision
TensorFlow
#deep-learning#image-segmentation#neural-network

black0017/MedicalZooPytorch

A PyTorch-based deep learning framework for 2D/3D medical image segmentation.

1.9K
Archived
Python
Computer Vision
API Frameworks
PyTorch
#medical-imaging#image-segmentation#deep-learning

YunYang1994/TensorFlow2.0-Examples

A collection of TensorFlow 2.0 examples for deep learning, computer vision, and more.

1.7K
Archived
Jupyter Notebook
Computer Vision
Frameworks
TensorFlow
#computer-vision#deep-learning#machine-learning

bubbliiiing/unet-pytorch

This is the source code for a PyTorch implementation of the U-Net model, which can be used to train custom models.

1.7K
Archived
Python
Computer Vision
PyTorch
#computer-vision#deep-learning#image-segmentation

yeyun111/dlcv_for_beginners

A library of code samples and examples for deep learning and computer vision, targeting beginners.

1.5K
Archived
Python
Computer Vision
Tutorials & Courses
PyTorch
#deep-learning#computer-vision#beginners

MIC-DKFZ/medicaldetectiontoolkit

A toolbox for training and deploying object detectors and segmentation models on medical images using PyTorch.

1.3K
Archived
Python
Computer Vision
API Frameworks
PyTorch
#medical-imaging#object-detection#segmentation

orobix/retina-unet

A Python library for segmenting retinal blood vessels using a convolutional neural network.

1.3K
Archived
Python
Computer Vision
Python
#computer-vision#image-segmentation#medical-imaging

The-AI-Summer/Deep-Learning-In-Production

This repository provides hands-on examples and guidance for building, training, deploying and maintaining deep learning models in production.

1.2K
Archived
Jupyter Notebook
ML Ops
API Frameworks
Python
#deep-learning#machine-learning#production

ternaus/TernausNet

A PyTorch implementation of the TernausNet model for image segmentation, pre-trained on the Kaggle Carvana dataset.

1.1K
Archived
Python
Computer Vision
Backend Frameworks
PyTorch
#image-segmentation#computer-vision#deep-learning
2

Stay in the loop

Get weekly updates on trending AI coding tools and projects.