Queries about the test output

[SayedNadim]

Hi, thanks for the good work. Can you please check whether this output is expected? output: input: mask:

The only change I made was the following code in the test script.

.....
mask = mask_transform(mask)
detail = img_transform(detail)
structure = img_transform(structure)
mask = torch.unsqueeze(mask, 0)
detail = torch.unsqueeze(detail, 0)
structure = torch.unsqueeze(structure, 0)

detail = detail * (1 - mask) # masking the details
structure = structure * (1 - mask) # masking the structure
......

Thanks!

[camellia120]

I am very glad to hear from you.Thank you very much!