VaticanCameos99 / knowledge-distillation-for-unet
An implementation of Knowledge distillation for segmentation, to train a small (student) UNet from a larger (teacher) UNet thereby reducing the size of the network while achieving performance similar to the heavier model.
☆52Updated 4 years ago
Alternatives and similar repositories for knowledge-distillation-for-unet:
Users that are interested in knowledge-distillation-for-unet are comparing it to the libraries listed below
- ☆134Updated 3 years ago
- Pytorch re-implementation of boundary loss, proposed in "Boundary Loss for Remote Sensing Imagery Semantic Segmentation"☆99Updated 4 years ago
- Decoder architecture based on the UNet++. Combining residual bottlenecks with depthwise convolutions and attention mechanisms, it outperf…☆66Updated 3 years ago
- draw ROC,PR curve and calculate AUC MAP IoU for image semantic segmentation problem☆70Updated 6 years ago
- Implementation of SETR model, Original paper: Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers.☆132Updated 4 years ago
- Official implementation of the Generalized Wasserstein Dice Loss in PyTorch☆80Updated 2 years ago
- An unofficial pytorch implementation for "Learning Active Contour Models for Medical Image Segmentation" by Chen, Xu, et al.☆75Updated 3 months ago
- ☆15Updated 2 years ago
- [ICCV 2021] "From Contexts to Locality: Ultra-high Resolution Image Segmentation via Locality-aware Contextual Correlation" by Qi Li, Wei…☆54Updated 3 years ago
- Official repository for Active Boundary Loss for Semantic Segmentation.☆73Updated 2 years ago
- A PyTorch implementation of Gated-SCNN based on ICCV 2019 paper "Gated-SCNN: Gated Shape CNNs for Semantic Segmentation"☆29Updated 4 years ago
- Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers☆196Updated 4 years ago
- Pytorch code for CVPR2021 paper "Learning Statistical Texture for Semantic Segmentation"☆94Updated 3 years ago
- [MICCAI 2021] BiX-NAS: Searching Efficient Bi-directional Architecture for Medical Image Segmentation☆38Updated 2 years ago
- [PR2022] The Code of “EANet: Iterative Edge Attention Network for Medical Image Segmentation”☆59Updated 2 years ago
- PyTorch implementation of Mobile-UNet.☆28Updated 2 years ago
- Implementation of Hausdorff loss function for DNN learning in segmentation tasks.☆65Updated 9 months ago
- [ICPR 2022] 3DConvCaps: 3DUnet with Convolutional Capsule Encoder for Medical Image Segmentation☆45Updated 2 years ago
- pytorch implementation of SEG-GRAD-CAM,which based on grad-cam☆51Updated 4 years ago
- Implementations of "Learning Euler's Elastica Model for Medical Image Segmentation"☆74Updated 4 years ago
- Pytorch implementation of FCN, UNet, PSPNet, and various encoder models.☆86Updated 4 years ago
- PyTorch implementation for Semantic Segmentation, include FCN, U-Net, SegNet, GCN, PSPNet, Deeplabv3, Deeplabv3+, Mask R-CNN, DUC, Google…☆55Updated 10 months ago
- Usage of Multi-task deep learning network for semantic segmentation in medical images☆131Updated 4 years ago
- Compound loss for PyTorch☆45Updated 2 years ago
- Using MobileNet as the backbone of UNet☆66Updated 4 years ago
- Meta Transfer Learning for Few Shot Semantic Segmentation using U-Net☆24Updated 4 years ago
- ☆53Updated 2 years ago
- A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder☆158Updated 4 years ago
- Code for Comprehensive Attention Convolutional Neural Networks for Explainable Medical Image Segmentation.☆176Updated 4 years ago
- Build U-Nets for segmentation from pre-trained TorchVision models.☆91Updated 5 years ago