MedSegDiff

MedSegDiff copied to clipboard

sample得到预测的噪声后怎么去除噪声呢？

5

我的任务是实现分割，输入的图片对是原图 分割mask图，在使用readme的教程中训练的我的代码，结果发现 模型参数ModelMeanType.EPSILON: noise,也就是unet预测的noise么，调用segmentation_sample.py函数得到的 sample, x_noisy, org, cal, cal_out = sample_fn( model, (batch_size, channels, args.image_size, args.image_size), img, step = args.diffusion_steps, clip_denoised=args.clip_denoised, model_kwargs=model_kwargs, ) sample是noise么，之后怎么处理得到我要的target分割图呢 我在可视化sample后发现和我想要的mask相差甚远，看了模型参数发现预测的是noise 值，是否有朋友知道有输入图 noise 变量 怎么得到mask图，请教各位互联网朋友们，感谢！

qxxfd

assert x_t.shape == eps.shape

16

你好，我在对于DRIVE数据集上进行采样时，出现了这样问题，请问我该如何解决呢 Traceback (most recent call last): File "E:\deep_learning\Segmentation\MedSegDiff-master\scripts\segmentation_sample.py", line 214, in main() File "E:\deep_learning\Segmentation\MedSegDiff-master\scripts\segmentation_sample.py", line 123, in main sample, x_noisy, org, cal, cal_out = sample_fn( File "E:\deep_learning\Segmentation\MedSegDiff-master\guided_diffusion\gaussian_diffusion.py", line 565, in...

2039551625

btcvloader

1

Thank you for all your contributions! But I have some questions: 1. When I read the code, I found that the script(btcvloader.py) is empty. Can you provide it(multi-organ segmentation)? 2....

dulicui742

Excuse me, I don't understand the multi-GPU training model using DP in segmentation_train.py and then using DDP in train_util.py. Is there a conflict?

ailinxiqingcheng

Help！RuntimeError: a leaf Variable that requires grad is being used in an in-place operation.

4

creating data loader... creating model and diffusion... training... Traceback (most recent call last): File "scripts/segmentation_train.py", line 118, in main() File "scripts/segmentation_train.py", line 70, in main TrainLoop( File "D:\MedSegDiff-master\.\guided_diffusion\train_util.py", line 83,...

doqizuo

how to use ddim if I should set time step

Hxxxz0

为什么训练出现了完全噪声的结果，如图所示的原图和标签的训练对，20000+图片对（单通道灰度图）训练200epoch得到如图结果，sample_step=30,是否有朋友知道是什么原因呢

1


qxxfd

Training on the Brats dataset yields errors, but can still be trained. Please tell me why these errors. /home/ubuntu/anaconda3/envs/medsegdiff/bin/python /media/ubuntu/89696b21-997d-42b2-a1c0-b16d972fd7b72/hlq/MedSegDiff-master/scripts/segmentation_train.py --data_name BRATS --data_dir /media/ubuntu/89696b21-997d-42b2-a1c0-b16d972fd7b72/hlq/data/BraTS2020/training --out_dir './BRATS2020_results/' --image_size 256 --num_channels 128...

1580827935

Thanks for your great work! I have a question regarding model efficiency mentioned in your paper. In Table 5 of the MedSegDiff-V2 paper, it states that the model has 46M...

YunduanLou

Hi, I'm having trouble using the code to train on my own dataset. The problem occurs with the Gaussian_Diffusion module. The question is as follows： Traceback (most recent call last):...

sonnystalin