-
Notifications
You must be signed in to change notification settings - Fork 186
Open
Description
我的任务是实现分割,输入的图片对是原图 分割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图,请教各位互联网朋友们,感谢!
Saint-lsy
Metadata
Metadata
Assignees
Labels
No labels