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.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

VaticanCameos99/knowledge-distillation-for-unet Stargazers