VaticanCameos99 / knowledge-distillation-for-unetLinks
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.
☆53Updated 5 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
Sorting:
- ☆138Updated 4 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
- draw ROC,PR curve and calculate AUC MAP IoU for image semantic segmentation problem☆70Updated 6 years ago
- A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder☆162Updated 4 years ago
- An unofficial pytorch implementation for "Learning Active Contour Models for Medical Image Segmentation" by Chen, Xu, et al.☆75Updated 8 months ago
- Pytorch implementation of FCN, UNet, PSPNet, and various encoder models.☆86Updated 5 years ago
- Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers☆197Updated 4 years ago
- Implementation of research paper : "PraNet: Parallel Reverse Attention Network for Polyp Segmentation" in Tensorflow☆21Updated 3 years ago
- Implementation of CVPR 2020 Dual Super-Resolution Learning for Semantic Segmentation☆104Updated 3 years ago
- Official repository for Active Boundary Loss for Semantic Segmentation.☆76Updated 2 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 code for CVPR2021 paper "Learning Statistical Texture for Semantic Segmentation"☆95Updated 4 years ago
- Usage of Multi-task deep learning network for semantic segmentation in medical images☆131Updated 5 years ago
- Pytorch re-implementation of boundary loss, proposed in "Boundary Loss for Remote Sensing Imagery Semantic Segmentation"☆100Updated 5 years ago
- [MICCAI 2021] BiX-NAS: Searching Efficient Bi-directional Architecture for Medical Image Segmentation☆38Updated 2 years ago
- Official PyTorch implementation of UACANet: Uncertainty Augmented Context Attention for Polyp Segmentation (ACMMM 2021)☆160Updated 2 years ago
- This repo is developed for evaluating binary image segmentation results. Measures, such as MAE, Precision, Recall, F-measure, PR curves a…☆133Updated 5 years ago
- Compound loss for PyTorch☆46Updated 2 years ago
- Implementation of SETR model, Original paper: Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers.☆133Updated 4 years ago
- The official implementation of "Rethink Dilated Convolution for Real-time Semantic Segmentation"☆87Updated last year
- PyTorch implementation for Semantic Segmentation, include FCN, U-Net, SegNet, GCN, PSPNet, Deeplabv3, Deeplabv3+, Mask R-CNN, DUC, Google…☆56Updated last year
- DDANet: Dual Decoder Attention Network for automatic Polyp Segmentation [International Conference on Pattern Recognition 2021]☆30Updated last year
- Introducing depthwise separable convolution into U-net in different ways.☆21Updated 4 years ago
- Decoder architecture based on the UNet++. Combining residual bottlenecks with depthwise convolutions and attention mechanisms, it outperf…☆66Updated 3 years ago
- Build U-Nets for segmentation from pre-trained TorchVision models.☆90Updated 6 years ago
- Repository containing code for "U-Net Fixed-Point Quantization for Medical Image Segmentation" paper at MICCAI2019☆84Updated 3 years ago
- Official implementation of the Generalized Wasserstein Dice Loss in PyTorch☆80Updated 3 years ago
- pytorch implementation of SEG-GRAD-CAM,which based on grad-cam☆52Updated 4 years ago
- Meta Transfer Learning for Few Shot Semantic Segmentation using U-Net☆24Updated 4 years ago
- ☆70Updated 5 years ago