flip 2 tensor mask with Pythorch

前端 未结 0 889
小蘑菇
小蘑菇 2021-02-15 09:11

I created 2 masks of 2 images by pytorch,

self.img_m = self.mask * self.img_truth
x = self.mask * self.img_m
y = (1 - self.mask) * self.img_g[-1].detach()
         


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题