code/train.py is the implementation of our method on the Prostate and Fundus dataset.

Modify the paths in lines 631 to 636 of the code.

code/train_MNMS is the implementation of our method on the M&Ms dataset.

The training and testing data should be in the format of the "data" folder.