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