From e7cf337dbac7fd49bb8f48ae3ca5e46fdabb0054 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 12 Oct 2020 10:23:03 -0600 Subject: [PATCH] Fix bug with chunk_with_reference --- codes/data/chunk_with_reference.py | 1 + 1 file changed, 1 insertion(+) diff --git a/codes/data/chunk_with_reference.py b/codes/data/chunk_with_reference.py index e066246c..ccb4a8b4 100644 --- a/codes/data/chunk_with_reference.py +++ b/codes/data/chunk_with_reference.py @@ -14,6 +14,7 @@ class ChunkWithReference: img = util.read_img(None, img_path, rgb=True) if img is None: return np.zeros(128, 128, 3) + return img def __getitem__(self, item): centers = torch.load(osp.join(self.path, "centers.pt"))