From ef2a27e4240e484d9ea6ee08453dc5a39e92d49b Mon Sep 17 00:00:00 2001 From: Trevor Spreadbury <30778843+trevorspreadbury@users.noreply.github.com> Date: Fri, 12 Jul 2019 11:38:13 -0500 Subject: [PATCH] Fixed image path to include /data subdir --- dataset/text_data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dataset/text_data.py b/dataset/text_data.py index 01d0997..079fd0d 100644 --- a/dataset/text_data.py +++ b/dataset/text_data.py @@ -29,8 +29,8 @@ def __getitem__(self, idx): name = self.config[self.mode][idx]["name"] text = self.config[self.mode][idx]["text"] - # img = cv2.imread(os.path.join(self.data_path, "data", name)) - img = cv2.imread(os.path.join(self.data_path, name)) + img = cv2.imread(os.path.join(self.data_path, "data", name)) + #img = cv2.imread(os.path.join(self.data_path, name)) seq = self.text_to_seq(text) sample = {"img": img, "seq": seq, "seq_len": len(seq), "aug": self.mode == "train"} if self.transform: