From 4e4a052d68ab08b4c630c508c9cb619a8f95947e Mon Sep 17 00:00:00 2001 From: "DHYANI\\DHYANI" Date: Wed, 27 Nov 2019 09:18:55 -0500 Subject: [PATCH] (bug fix) data type incompatibility between boolean values while creating instance_edge_map --- models/pix2pix_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/pix2pix_model.py b/models/pix2pix_model.py index 784cb15a..5ec24b1b 100644 --- a/models/pix2pix_model.py +++ b/models/pix2pix_model.py @@ -235,7 +235,7 @@ def divide_pred(self, pred): return fake, real def get_edges(self, t): - edge = self.ByteTensor(t.size()).zero_() + edge = self.ByteTensor(t.size()).zero_().bool() edge[:, :, :, 1:] = edge[:, :, :, 1:] | (t[:, :, :, 1:] != t[:, :, :, :-1]) edge[:, :, :, :-1] = edge[:, :, :, :-1] | (t[:, :, :, 1:] != t[:, :, :, :-1]) edge[:, :, 1:, :] = edge[:, :, 1:, :] | (t[:, :, 1:, :] != t[:, :, :-1, :])