Clone/DIS
1
0
Fork 0
mirror of https://github.com/xuebinqin/DIS.git synced 2024-11-29 18:23:08 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

update data_loader_cache.py

Browse Source
This commit is contained in:
PiggyJerry 2022-07-27 23:54:20 +04:00
parent 6e807f8d10
commit db68ff3480
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
IS-Net/data_loader_cache.py
View File

@ -76,7 +76,7 @@ def create_dataloaders(name_im_gt_list, cache_size=[], cache_boost=True, my_tran
gos_dataloaders = [] gos_dataloaders = []
gos_datasets = [] gos_datasets = []
# if(mode=="train"):
if(len(name_im_gt_list)==0): if(len(name_im_gt_list)==0):
return gos_dataloaders, gos_datasets return gos_dataloaders, gos_datasets