当前位置：网站首页>get_ edges

get_ edges

2022-06-23 16:15:00 【Growth of code Xiaobai】

 def get_edges(self, t):
        torchvision.transforms.ToPILImage()(t[0].cpu()).show()
        edge = self.ByteTensor(t.size()).zero_()
        edge[:, :, :, 1:] = edge[:, :, :, 1:] | (t[:, :, :, 1:] != t[:, :, :, :-1])
        edge[:, :, :, :-1] = edge[:, :, :, :-1] | (t[:, :, :, 1:] != t[:, :, :, :-1])
        edge[:, :, 1:, :] = edge[:, :, 1:, :] | (t[:, :, 1:, :] != t[:, :, :-1, :])
        edge[:, :, :-1, :] = edge[:, :, :-1, :] | (t[:, :, 1:, :] != t[:, :, :-1, :])
        torchvision.transforms.ToPILImage()(edge.float()[0].cpu()).show()
        return edge.float()