diffusion-anomaly icon indicating copy to clipboard operation
diffusion-anomaly copied to clipboard

Train/Test Dataset Split and Preprocessing

Open tungnthust opened this issue 1 year ago • 4 comments

I want to reproduce your results experimented on BRATS20 dataset reported in your paper. However, I have some troubles in processing that dataset. Could you share the way you split train/test dataset (may be a list of patient ids for each set) and how you normalize raw data into range between 0 and 1 ?

tungnthust avatar Apr 14 '23 07:04 tungnthust

Me too. I need the split list as well

ChyaZhang avatar Jun 01 '23 07:06 ChyaZhang

@JuliaWolleb Could you kindly share hyperparmeters for training classifier for BRATS dataset (i.e learning_rate, batch_size, anneal_lr, weight_decay, dropout) ? I tried your training settings in README, but it does not produce good results (much worse than your pretrained classifier checkpoint you provided). I think the hyperparameters in TRAINING_FLAGS are used for just training diffusion model, not for the classifier model. Since I refered to classifer hyperparmeters used in openai/guided_diffusion repository, it is completely different from your settings cited in README.

tungnthust avatar Jun 06 '23 00:06 tungnthust

yes, for training the classifier, you need the CLASSIFIER_FLAGS and TRAIN_FLAGS of the README file. The default values are given in the classifier_defaults https://github.com/JuliaWolleb/diffusion-anomaly/blob/509643c3c15fc5af7d4712aa7b1e8ba8a06214ff/guided_diffusion/script_util.py#L28 Then it should work fine.

JuliaWolleb avatar Jun 20 '23 07:06 JuliaWolleb

Hi, I wrote a data set preprocessing code now, you can see if it can be used, please visit: https://github.com/JuliaWolleb/diffusion-anomaly/issues/19

waiterxiaoyy avatar Aug 14 '23 06:08 waiterxiaoyy