Open max0312 opened 1 year ago
Hi
Hi, Thank you for publishing your work. Im trying to reproduce your work for a week now and couldn't figure out what's wrong with my reproduction. After I trained the classifier and diffusion model, I run the classifier_sample_known.py with checkpoint brats2update050000.pt and modelbratsclass020000.pt. However, I got the results as below. The sampled_output is just the black and white image. I didn't change anything from your code. Could you figure what I'm doing wrong here? Thank you.
can you run the script classifier_sample_known.py with a classifier_scale of 0? then the output should show the same image as the input.
I still got output as back and white image. What do you think of in this situation? The input is clearly correct, but the sample output is wrong.
Do you mind uploading your checkpoints?
Thanks for your advice and I rechecked the results
Thank you very much for your sharing but I encountered a similar problem. I used the checkpoints in your link above, but still output black and white images. And there is no obvious noise on reversesample image. For the code, I just modified bratsloader.py. Can you give me some advice?
I still got output as back and white image. What do you think of in this situation? The input is clearly correct, but the sample output is wrong.
hello, have you solved this problem after using the checkpoints uploaded?
There is something going wrong with your noise encoding and decoding. How large did you choose L? The image "reversesample" should show a noisy image of noise level L. If you put s=0, the final images should show the same images as the input. Could you check four function "ddim_sample_loop_known"?
Hi, thanks for your great job I have some questions after training and testing on the brats dataset
Did you normalize the seg part(data/brats/training/000001/brats_train_001_seg_080_w.nii.gz) during preprocessing? I see that the training sample set you gave has not been normalized. Will this affect the cross_entropy calculation during classifier training?
I found that the results generated by using the brats2update010000.pt and modelbratsclass010000.pt saved at the beginning of the training are better than those of the model saved in the later training. What is the reason for this? (What are the specific model parameters you used in the test?)