unet-pytorch

There are 202 repositories under unet-pytorch topic.

  • arshadowgan-like

    ARShadowGAN-like realization. PyTorch.

    Language:Python31

  • pytorch-UNET

    U-Net: Convolutional Networks for Biomedical Image Segmentation

    Language:Python31

  • LungSegmentation

    This was my research project at IIT Bombay on Lung Segmentation from Chest X-Rays Images

    Language:Python31

  • unet_pytorch

    Pytorch implementation of UNet for converting aerial satellite images into google maps kinda images.

    Language:Python30

  • CV-in-ADAS-pytorch

    This repo includes Unet, Spatial CNN (S-CNN) and VPNet for lane segmentation, and YOLO, Faster-RCNN, Stereo-RCNN for vehicle detection.

    Language:Jupyter Notebook28

  • U-Net-in-PyTorch

    This is an implementation of the U-Net model from the paper, U-Net: Convolutional Networks for Biomedical Image Segmentation.

    Language:Jupyter Notebook26

  • BeamDosePrediction

    A python (Pytorch) implementation of Beam Dose Decomposition for Dose Prediction [MICCAI 2022]

    Language:Python26

  • U-Net---Biomedical-Image-Segmentation

    Implementation of the paper titled - U-Net: Convolutional Networks for Biomedical Image Segmentation @ https://arxiv.org/abs/1505.04597

    Language:Jupyter Notebook26

  • RSANet

    RSANet: Recurrent Slice-wise Attention Network for Multiple Sclerosis Lesion Segmentation (MICCAI 2019)

    Language:Python26

  • MTL-Segmentation

    Meta Transfer Learning for Few Shot Semantic Segmentation using U-Net

    Language:Python24

  • Kaggle-Steel-Defect-Detection

    Code of kaggle semantic segmentation competition: Steel Defect Detection.

    Language:Python24

  • Retinal-vessel-segmentation

    :tada:一个基于 UNet 的视网膜血管分割项目，使用 PyTorch 实现并基于 DRIVE 数据集进行训练和测试。项目包括完整的数据处理、模型训练和测试流程，最终生成视网膜图像的分割结果。

    Language:Python22

  • LungTumor-Segmentation

    Automatically segment lung cancer in CTs

    Language:Jupyter Notebook19

  • Salt-Segmentation-UNet

    Implement a salt segmentation using PyTorch for TGS Salt Identification Challenge

    Language:Python16

  • resnet34-unet

    Segmentation model using UNET architecture with ResNet34 as encoder background, designed with PyTorch.

    Language:Python16

  • Hippocampus_Segmentation_MRI

    Udacity AI for Healthcare Nanodegree Project: Measurement of Hippocampus Structure in MRI 3-D Images using Deep Learning Image Segmentation

    Language:Jupyter Notebook16

  • Image-Segmentation-PyTorch

    U-Net for image segmentation, PyTorch implementation.

    Language:Python16

  • pytorch-face-reaging-network

    PyTorch Lightning implementation of Disney's face re-aging network (FRAN) paper.

    Language:Python15

  • camus-segmentation-pytorch

    PyTorch implementation of the two U-Net-based architectures described in "Deep Learning for Segmentation using an Open Large-Scale Dataset in 2D Echocardiography"

    Language:Python15

  • CamVid-Segmentation-Pytorch

    This is the DL repository for Semantic Segmentation using U-Net model in pytorch library.

    Language:Jupyter Notebook15

  • leibniz

    Leibniz is a python package which provide facilities to express learnable partial differential equations with PyTorch

    Language:Python15

  • PyTorch-Unet

    PyTorch1.0 Implementation of Unet

    Language:Python15

  • lane-detection

    Weakly-supervised road-lane markings detection for autonomous driving, mitigating the lack of training data

    Language:Python14

  • UNet

    PyTorch implementation of Original UNet Paper

    Language:Python13

  • PyTorch-UNet

    PyTorch implementation of the UNet model -- https://arxiv.org/abs/1505.04597

    Language:Python13

  • UNet-MISO

    unofficial implementation of "A Causal U-net based Neural Beamforming Network for Real-Time Multi-Channel Speech Enhancement"

    Language:Python12

  • cospgd

    The official repository for CosPGD: a unified white-box adversarial attack for pixel-wise prediction tasks.

    Language:Python12

  • Pytorch-UNet-Brain-Tumor-Detection

    simple pytorch unet model for brain tumor detection on MRI tiff images

    Language:Python12

  • VesselExtract

    U-net based CNN for segmenting blood vessel and thereafter removal of vessels from fundus image

    Language:Jupyter Notebook12

  • Unet_DWT

    Unet based on Wavelet coefficients for segmentation

    Language:Python11

  • UNET-Semantic_Segmentation

    Semantic segmentation on CamVid dataset using the U-Net.

    Language:Jupyter Notebook9
  • Pytorch-UNet-2

    Pytorch-UNet-2

    UNet 2.0 - Pytorch implementation of the U-Net 2.0 for image semantic segmentation, with processing blocks for noisy images

    Language:Python9

  • Dynamic-Unet

    This project is the implementation of Dynamic U-Net architecture on Caravan Mask Challenge Dataset. A state of the art technique that has won many Kaggle competitions and is widely used in industry. Image segmentation models allow us to precisely classify every part of an image, right down to pixel level.

    Language:Jupyter Notebook9

  • pannuke-segmentation

    An open-source UNet-based pipeline for nuclei segmentation in histopathology images using the PanNuke dataset. It features an interactive web app for easy data visualization and handling, making AI tools accessible even for non-experts. This project provides a foundation for training and exploring histopathology data.

    Language:Python8

  • Doubleunet_pytorch

    Official implementation of DoubleU-Net: A Deep Convolutional Neural Network for Medical Image Segmentation (pytorch implementation)

    Language:Python8

  • TransRUPNet

    TransRUPNet for Improved Out-of-Distribution Generalization in Polyp Segmentation (IEEE EMBC)

    Language:Python7