为什么将Ground Truth与Image连接在一起作为模型的input进行训练？

JuliaWolleb / Diffusion-based-Segmentation

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".

MIT License

278 stars 36 forks source link

为什么将Ground Truth与Image连接在一起作为模型的input进行训练？ #62

Closed ZhengChen6 closed 6 months ago

ZhengChen6 commented 6 months ago

您好，在阅读您的代码的时候，我发现您将标签与图像连接在一起送入模型，这样训练的话模型岂不是在训练过程中就看见了标签？如下面您的代码中，mask是从x_start中得到的，说明x_start中包含了标签mask，并且我通过调试也发现x_start的shape是 (10,5,224,224) 。我不太理解将mask也送入模型进行训练的原因，请问这该怎么理解，谢谢！

ZhengChen6 commented 6 months ago

已经明白了，谢谢！