The forward method of InpaintingModel is as follows:
def forward(self, images, edges, masks):
images_masked = (images * (1 - masks).float()) + masks
inputs = torch.cat((images_masked, edges), dim=1)
outputs = self.generator(inputs)
return outputs
I visualized the variables about the image. As shown below:
I think there is a problem with my images_masked variable. Is this input correct?
The forward method of InpaintingModel is as follows: def forward(self, images, edges, masks): images_masked = (images * (1 - masks).float()) + masks inputs = torch.cat((images_masked, edges), dim=1) outputs = self.generator(inputs) return outputs I visualized the variables about the image. As shown below:
I think there is a problem with my images_masked variable. Is this input correct?