Projects sharing features with Segmentation Models.pytorch

Detectron2 is a PyTorch computer vision framework and visual recognition platform designed for training and deploying models for object detection, image segmentation, and visual recognition. It provides a research-oriented environment for training complex vision models with multi-GPU acceleration. The project includes a specialized object detection library for identifying and locating multiple objects via bounding boxes, as well as an image segmentation toolkit for creating pixel-level masks through instance, semantic, and panoptic segmentation. Additionally, it features a human pose estimati

Official Pytorch Code base for "Multi-Level Global Context Cross Consistency Model for Semi-Supervised Ultrasound Image Segmentation with Diffusion Model"

Diff-UNet: A Diffusion Embedded Network for Volumetric Segmentation. (using diffusion for 3D medical image segmentation)

DALL-E for Detection: Language-driven Compositional Image Synthesis for Object Detection

Data & Code for FEDD published @ MICCAI 23

Code for paper DMCVR: Morphology-Guided Diffusion Model for 3D Cardiac Volume Reconstruction

Offcial PyTorch repository for the paper "Generative Semantic Communication: Diffusion Models Beyond Bit Recovery"

IJCV 2024 MosaicFusion: Diffusion Models as Data Augmenters for Large Vocabulary Instance Segmentation

Repository for "Stochastic Segmentation with Conditional Categorical Diffusion Models" (ICCV 2023)

A JAX-based deep learning framework for image segmentation using diffusion models.

MICCAI 2023 DermoSegDiff: A Boundary-aware Segmentation Diffusion Model for Skin Lesion Delineation

Nvidia Semantic Segmentation monorepo

Remote Sensing Change Detection using Denoising Diffusion Probabilistic Models

NeurIPS 2023: Towards Generic Semi-Supervised Framework for Volumetric Medical Image Segmentation

Label-Efficient Semantic Segmentation with Diffusion Models (ICLR'2022)

Personalize Segment Anything Model (SAM) with 1 shot in 10 seconds