jeya-maria-jose / KiU-Net-pytorch

Official Pytorch Code of KiU-Net for Image/3D Segmentation - MICCAI 2020 (Oral), IEEE TMI
https://sites.google.com/view/kiunet/kiu-net
MIT License
357 stars 80 forks source link

i read your paper, you said the params of kiunet is much small than unet cause you only have 32,64,128 conv. but the kiunet in model.py seems not like that,why? #11

Closed jiangjiaxi20 closed 3 years ago

jeya-maria-jose commented 3 years ago

The kiunet 2D that we use for image segmentation experiments does have 32,64,128 conv. For LITS dataset, in model.py we in fact have 16,32,64 number of filters which are less because the input was larger in size compared to 2D experiments and we wanted to release a code that was able to fit in a 12 GB GPU. Please feel free to increase the number of filters if memory is not an issue.