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
- ☆135Updated 3 years ago
- A PyTorch implementation of Gated-SCNN based on ICCV 2019 paper "Gated-SCNN: Gated Shape CNNs for Semantic Segmentation"☆28Updated 4 years ago
- Decoder architecture based on the UNet++. Combining residual bottlenecks with depthwise convolutions and attention mechanisms, it outperf…☆66Updated 3 years ago
- Repo for ICPR2020 paper "Crack Detection as a Weakly-Supervised Problem: Towards Achieving Less Annotation-Intensive Crack Detectors"☆17Updated 4 years ago
- Pytorch re-implementation of boundary loss, proposed in "Boundary Loss for Remote Sensing Imagery Semantic Segmentation"☆99Updated 4 years ago
- Pytorch code for CVPR2021 paper "Learning Statistical Texture for Semantic Segmentation"☆94Updated 3 years ago
- pytorch implementation of SEG-GRAD-CAM,which based on grad-cam☆51Updated 4 years ago
- Implementation of SETR model, Original paper: Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers.☆132Updated 4 years ago
- An unofficial pytorch implementation for "Learning Active Contour Models for Medical Image Segmentation" by Chen, Xu, et al.☆75Updated 4 months ago
- ☆15Updated 2 years ago
- [PR2022] The Code of “EANet: Iterative Edge Attention Network for Medical Image Segmentation”☆59Updated 2 years ago
- Official repository for Active Boundary Loss for Semantic Segmentation.☆74Updated 2 years ago
- Usage of Multi-task deep learning network for semantic segmentation in medical images☆130Updated 4 years ago
- draw ROC,PR curve and calculate AUC MAP IoU for image semantic segmentation problem☆70Updated 6 years ago
- Compound loss for PyTorch☆45Updated 2 years ago
- Applying Octave UNet for Retinal Vessel Segmentation.☆22Updated 4 years ago
- ☆111Updated 3 years ago
- Pytorch implementation of FCN, UNet, PSPNet, and various encoder models.☆86Updated 4 years ago
- FDDWNET: A LIGHTWEIGHT CONVOLUTIONAL NEURAL NETWORK FOR REAL-TIME SEMANTIC SEGMENTATION(ICASSP2020)☆9Updated 4 years ago
- DDANet: Dual Decoder Attention Network for automatic Polyp Segmentation [International Conference on Pattern Recognition 2021]☆28Updated last year
- Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers☆196Updated 4 years ago
- Official implementation of the Generalized Wasserstein Dice Loss in PyTorch☆80Updated 2 years ago
- Official implementation of ResUNet++, CRF, and TTA for segmentation of medical images (IEEE JBIHI)☆124Updated last year
- The official implementation of "Rethink Dilated Convolution for Real-time Semantic Segmentation"☆87Updated last year
- Scale-ware Strip Attention Guided Feature Pyramid Network for Real-time Semantic Segmentation☆21Updated last year
- A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder☆158Updated 4 years ago
- ☆52Updated 4 months ago
- use FPN to FPN-Semantic-segmentation。pytorch version☆50Updated 5 years ago
- Introducing depthwise separable convolution into U-net in different ways.☆20Updated 4 years ago
- PyTorch implementation of Mobile-UNet.☆28Updated 2 years ago