search

TFboys-lzz / MPSCL

This repository contains code for the paper "Margin Preserving Self-paced Contrastive Learning Towards Domain Adaptation for Medical Image Segmentation", published at IEEE JBHI 2022
29 stars 1 forks source link

pretrain on source domain #7

Closed wh-isper closed 1 year ago

wh-isper commented 1 year ago

Dear author, I have a question. If i want to run train_UDA.py to train MPSCL, I need a pretrained model on source domain which you have provided, but you did not provide the code to pretrain model on source data?

wh-isper commented 1 year ago

If I want to train on my own dataset from scratch, should I write a code to pretrain on source domain or just run train_UDA.py to get pretrained model and run train_UDA again to get final model?

TFboys-lzz commented 1 year ago

I suggest you rewrite the code to pre-train only in the source domain so as to generate the corresponding pseudo-labels for the target domain data.

wh-isper commented 1 year ago

I got it!