search

ChongQingNoSubway / SelfReg-UNet

Code for the paper "SelfReg-UNet: Self-Regularized UNet for Medical Image Segmentation "
MIT License
36 stars 5 forks source link

3D KD_Loss #7

Closed wakakaFF closed 2 months ago

wakakaFF commented 2 months ago

Thanks for your work!

I want to use it in 3D segmentation, so I made some changes to the code, but the experiment didn't improve much.

Is there anything in my code that hasn't been changed?

3c246b6f2d4a6c17087540d0cf0c04e f932ea1b26609f0cde5a67a35959058

ChongQingNoSubway commented 2 months ago

Thanks for being interested. The inter_fd calculation should include features from the encoder parts, and the decoders need to consider the different blocks at each stage (e.g., Each downsample stage has two transformer blocks SwimUnet ).