I think there are some errors in models.py code (in the forward functions in both models).
edges_masked = (edges (1 - masks))
images_masked = (images (1 - masks)) + masks
I used the debugger to make sure the values are correct and also I wrote the images in disk. The code should be:
edges_masked = edges masks
images_masked = images masks
I may be wrong, but please check it.
Dear sir, Thank you for your great work.
I think there are some errors in models.py code (in the forward functions in both models). edges_masked = (edges (1 - masks)) images_masked = (images (1 - masks)) + masks I used the debugger to make sure the values are correct and also I wrote the images in disk. The code should be: edges_masked = edges masks images_masked = images masks I may be wrong, but please check it.
Thank you.