From 362f4e969990040f8f21565dddffcafa954690be Mon Sep 17 00:00:00 2001 From: Xuebin Qin Date: Tue, 26 Jul 2022 19:55:35 -0700 Subject: [PATCH] Update data_loader_cache.py temporarily solution for training bugs --- IS-Net/data_loader_cache.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/IS-Net/data_loader_cache.py b/IS-Net/data_loader_cache.py index ebcc038..e7e3a13 100644 --- a/IS-Net/data_loader_cache.py +++ b/IS-Net/data_loader_cache.py @@ -291,8 +291,8 @@ class GOSDatasetCache(Dataset): gt, gt_shp = gt_preprocess(gt,self.cache_size) gt_cache_file = os.path.join(cache_folder,self.dataset["data_name"][i]+"_"+im_id + "_gt.pt") torch.save(gt,gt_cache_file) - # cached_dataset["gt_path"][i] = gt_cache_file - cached_dataset["gt_path"].append(gt_cache_file) + cached_dataset["gt_path"][i] = gt_cache_file + #cached_dataset["gt_path"].append(gt_cache_file) if(self.cache_boost): gts_pt_list.append(torch.unsqueeze(gt,0)) # gts_list.append(gt.cpu().data.numpy().astype(np.uint8))