Submitted by Hulord t3_y4u0zl in MachineLearning

Hello everyone,

I am currently starting to do research on using contrastive learning for medical image segmentation. From all the papers I have read, I am trying to use SimCLR as pretraining for image segmentation, since I am dealing with medical images I will use the Unet for the downstream task.

My question is how do I use the SimCLR as pretraining and is there any good code baseline for this?

11

Comments

You must log in or register to comment.

Hulord OP t1_isth4xm wrote

I´m having a bit of a struggle trying to pre-train using my data. I'm using a MedMNIST dataset (https://medmnist.com), specifically the BreastMNIST. I have the data in a DataLoader but can´t really understand how to load this dataloader to the configuration of the model by reading the documentation. Does it only work by having a downloaded dataset?

1