IS_Net/datalist.py

dataset_test = {"name": "DIS5K-test",
                "im_dir": r"DIS5K/DIS5K-test/im",
                "gt_dir": r"DIS5K/DIS5K-test/gt",
                "mid_dir":r"DIS5K/DIS5K-test/mask",
                "im_ext": ".jpg",
                "gt_ext": ".png",
                "mid_ext": ".png",
                "cache_dir":r"DIS5K-Cache/DIS-test"}

dataset_tr = {"name": "DIS5K-TR-m",
                "im_dir": r"DIS5K/DIS-TR/im",
                "gt_dir": r"DIS5K/DIS-TR/gt",
                "mid_dir":r"DIS5K-TR/mask",
                "im_ext": ".jpg",
                "gt_ext": ".png",
                "mid_ext": ".png",
                "cache_dir":r"DIS5K-Cache/DIS-TR-m"}

dataset_vd = {"name": "DIS5K-VD-m",
                "im_dir": r"DIS5K/DIS-VD/im",
                "gt_dir": r"DIS5K/DIS-VD/gt",
                "mid_dir":r"DIS5K/DIS5K-VD/mask",
                "im_ext": ".jpg",
                "gt_ext": ".png",
                "mid_ext": ".png",
                "cache_dir":r"DIS5K-Cache/DIS-VD-m"}

dataset_te1 = {"name": "DIS5K-TE1-m",
                "im_dir": r"DIS5K/DIS-TE1/im",
                "gt_dir": r"DIS5K/DIS-TE1/gt",
                "mid_dir":r"DIS5K/DIS5K-TE1/mask",
                "im_ext": ".jpg",
                "gt_ext": ".png",
                "mid_ext": ".png",
                "cache_dir":r"DIS5K-Cache/DIS-TE1-m"}

dataset_te2 = {"name": "DIS5K-TE2-m",
                "im_dir": r"DIS5K/DIS-TE2/im",
                "gt_dir": r"DIS5K/DIS-TE2/gt",
                "mid_dir":r"DIS5K/DIS5K-TE2/mask",
                "im_ext": ".jpg",
                "gt_ext": ".png",
                "mid_ext": ".png",
                "cache_dir":r"DIS5K-Cache/DIS-TE2-m"}

dataset_te3 = {"name": "DIS5K-TE3-m",
                "im_dir": r"DIS5K/DIS-TE3/im",
                "gt_dir": r"DIS5K/DIS-TE3/gt",
                "mid_dir":r"DIS5K/DIS5K-TE3/mask",
                "im_ext": ".jpg",
                "gt_ext": ".png",
                "mid_ext": ".png",
                "cache_dir":r"DIS5K-Cache/DIS-TE3-m"}

dataset_te4 = {"name": "DIS5K-TE4-m",
                "im_dir": r"DIS5K/DIS-TE4/im",
                "gt_dir": r"DIS5K/DIS-TE4/gt",
                "mid_dir":r"DIS5K/DIS5K-TE4/mask",
                "im_ext": ".jpg",
                "gt_ext": ".png",
                "mid_ext": ".png",
                "cache_dir":r"DIS5K-Cache/DIS-TE4-m"}