search

FengheTan9 / Multi-Level-Global-Context-Cross-Consistency

Official Pytorch Code base for "Multi-Level Global Context Cross Consistency Model for Semi-Supervised Ultrasound Image Segmentation with Diffusion Model"
MIT License
33 stars 1 forks source link

pretrained model #5

Open SuroshAhmadZobair opened 1 year ago

SuroshAhmadZobair commented 1 year ago

Hi Thanks for your work on this repo!

What would you recommend to use as the pre-trained model? Have you used and seen an improvement if finetuning a pre-trained model?

Thanks in advance

FengheTan9 commented 1 year ago

Thanks for using our code. U-Net is recommend. It is a pyramid encoder-decoder structure with skip-connections and is very suitable for medical image segmentation tasks with scarce data. We did not use pre-trained weights and we trained from scratch.