File size: 24,579 Bytes
f7f604d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 |
import os
import sys
import argparse
filepath = os.path.split(os.path.abspath(__file__))[0]
repopath = os.path.split(filepath)[0]
sys.path.append(repopath)
def download_and_unzip(filename, url, dest, unzip=True, **kwargs):
if not os.path.isdir(dest):
os.makedirs(dest, exist_ok=True)
os.system("wget -O {} {}".format(os.path.join(dest, filename), url))
if unzip:
os.system("unzip -o {} -d {}".format(os.path.join(dest, filename), dest))
os.system("rm {}".format(os.path.join(dest, filename)))
def _args():
parser = argparse.ArgumentParser()
parser.add_argument('-d', '--dest', type=str, default=None)
parser.add_argument('-e', '--extra', action='store_true', default=False)
return parser.parse_args()
train_datasets = [
{'filename': "DUTS-TR.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EQ7L2XS-5YFMuJGee7o7HQ8BdRSLO8utbC_zRrv-KtqQ3Q\?e\=bCSIeo\&download\=1", 'dest': "data/Train_Dataset", 'unzip': True, 'extra': False},
{'filename': "HRSOD-TR.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EfUx92hUgZJNrWPj46PC0yEBXcorQskXOCSz8SnGH5AcLQ\?e\=WA5pc6\&download\=1", 'dest': "data/Train_Dataset", 'unzip': True, 'extra': True},
{'filename': "UHRSD-TR.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/Ea4_UCbsKmhKnMCccAJOTLgBmQFsQ4KhJSf2jx8WQqj3Wg\?e\=18kYZS\&download\=1", 'dest': "data/Train_Dataset", 'unzip': True, 'extra': True},
{'filename': "DIS-TR.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EZtZJ493tVNJjBIpNLdus68B3u906PdWtHsf87pulj78jw\?e\=bUg2UQ\&download\=1", 'dest': "data/Train_Dataset", 'unzip': True, 'extra': True}
]
test_datasets = [
{'filename': "DUTS-TE.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EfuCxjveXphPpIska9BxHDMBHpYroEKdVlq9HsonZ4wLDw\?e\=Mz5giA\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': False},
{'filename': "DUT-OMRON.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ERvApm9rHH5LiR4NJoWHqDoBCneUQNextk8EjQ_Hy0bUHg\?e\=wTRZQb\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': False},
{'filename': "ECSSD.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ES_GCdS0yblBmnRaDZ8xmKQBPU_qeECTVB9vlPUups8bnA\?e\=POVAlG\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': False},
{'filename': "HKU-IS.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EYBRVvC1MJRAgSfzt0zaG94BU_UWaVrvpv4tjogu4vSV6w\?e\=TKN7hQ\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': False},
{'filename': "PASCAL-S.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EfUDGDckMnZHhEPy8YQGwBQB5MN3qInBkEygpIr7ccJdTQ\?e\=YarZaQ\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': False},
{'filename': "DAVIS-S.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/Ebam8I2o-tRJgADcq-r9YOkBCDyaAdWBVWyfN-xCYyAfDQ\?e\=Mqz8cK\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': False},
{'filename': "HRSOD-TE.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EbHOQZKC59xIpIdrM11ulWsBHRYY1wZY2njjWCDFXvT6IA\?e\=wls17m\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': False},
{'filename': "UHRSD-TE.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EUpc8QJffNpNpESv-vpBi40BppucqOoXm_IaK7HYJkuOog\?e\=JTjGmS\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': False},
{'filename': "FSS-1000.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EaP6DogjMAVCtTKcC_Bx-YoBoBSWBo90lesVcMyuCN35NA\?e\=0DDohA\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': True},
{'filename': "MSRA-10K.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EauXsIkxBzhDjio6fW0TubUB4L7YJc0GMTaq7VfjI2nPsg\?e\=c5DIxg\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': True},
{'filename': "DIS-VD.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EYJm3BqheaxNhdVoMt6X41gBgVnE4dulBwkp6pbOQtcIrQ\?e\=T6dtXm\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': True},
{'filename': "DIS-TE1.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EcGYE_Gc0cVHoHi_qUtmsawB_5v9RSpJS5PIAPlLu6xo9A\?e\=Nu5mkQ\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': True},
{'filename': "DIS-TE2.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EdhgMdbZ049GvMv7tNrjbbQB1wL9Ok85YshiXIkgLyTfkQ\?e\=mPA6Po\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': True},
{'filename': "DIS-TE3.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EcxXYC_3rXxKsQBrp6BdNb4BOKxBK3_vsR9RL76n7YVG-g\?e\=2M0cse\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': True},
{'filename': "DIS-TE4.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EdkG2SUi8flJvoYbHHOmvMABsGhkCJCsLLZlaV2E_SZimA\?e\=zlM2kC\&download\=1", 'dest': "data/Test_Dataset", 'unzip': True, 'extra': True}
]
backbone_ckpts = [
{'filename': "res2net50_v1b_26w_4s-3cf99910.pth" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EUO7GDBwoC9CulTPdnq_yhQBlc0SIyyELMy3OmrNhOjcGg\?e\=T3PVyG\&download\=1", 'dest': "data/backbone_ckpt", 'unzip': False, 'extra':False},
{'filename': "swin_base_patch4_window12_384_22kto1k.pth" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ESlYCLy0endMhcZm9eC2A4ABatxupp4UPh03EcqFjbtSRw\?e\=7y6lLt\&download\=1", 'dest': "data/backbone_ckpt", 'unzip': False, 'extra':False}
]
pretrained_ckpts = [
{'filename': "latest.pth", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ET0R-yM8MfVHqI4g94AlL6AB-D6LxNajaWeDV4xbVQyh7w\?e\=l4JkZn\&download\=1", 'dest': "snapshots/Plus_Ultra" , 'unzip': False, 'extra':True},
]
precomputed_maps = [
{'filename': "DUTS-TE.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/Eb0iKXGX1vxEjPhe9KGBKr0Bv7v2vv6Ua5NFybwc6aIi1w\?e\=oHnGyJ\&download\=1", 'dest': "snapshots/InSPyReNet_Res2Net50", 'unzip': True, 'extra': False},
{'filename': "DUT-OMRON.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/Ef1HaYMvgh1EuuOL8bw3JGYB41-yo6KdTD8FGXcFZX3-Bg\?e\=TkW2m8\&download\=1", 'dest': "snapshots/InSPyReNet_Res2Net50", 'unzip': True, 'extra': False},
{'filename': "ECSSD.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EdEQQ8o-yI9BtTpROcuB_iIBFSIk0uBJAkNyob0WI04-kw\?e\=cwEj2V\&download\=1", 'dest': "snapshots/InSPyReNet_Res2Net50", 'unzip': True, 'extra': False},
{'filename': "HKU-IS.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/Ec6LyrumVZ9PoB2Af0OW4dcBrDht0OznnwOBYiu8pdyJ4A\?e\=Y04Fmn\&download\=1", 'dest': "snapshots/InSPyReNet_Res2Net50", 'unzip': True, 'extra': False},
{'filename': "PASCAL-S.zip" , 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ETPijMHlTRZIjqO5H4LBknUBmy8TGDwOyUQ1H4EnIpHVOw\?e\=k1afrh\&download\=1", 'dest': "snapshots/InSPyReNet_Res2Net50", 'unzip': True, 'extra': False},
{'filename': "DUTS-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ETumLjuBantLim4kRqj4e_MBpK_X5XrTwjGQUToN8TKVjw\?e\=ZT8AWy\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB", 'unzip': True, 'extra': False},
{'filename': "DUT-OMRON.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EZbwxhwT6dtHkBJrIMMjTnkBK_HaDTXgHcDSjxuswZKTZw\?e\=9XeE4b\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB", 'unzip': True, 'extra': False},
{'filename': "ECSSD.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ESfQK-557uZOmUwG5W49j0EBK42_7dMOaQcPsc_U1zsYlA\?e\=IvjkKX\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB", 'unzip': True, 'extra': False},
{'filename': "HKU-IS.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EURH96JUp55EgUHI0A8RzKoBBqvQc1nVb_a67RgwOY7f-w\?e\=IP9xKa\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB", 'unzip': True, 'extra': False},
{'filename': "PASCAL-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EakMpwONph9EmnCM2rS3hn4B_TL42T6tuLjBEeEa5ndkIw\?e\=XksfA5\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB", 'unzip': True, 'extra': False},
{'filename': "DAVIS-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ETUCKFX0k8lAvpsDj5sT23QB2ohuE_ST7oQnWdaW7AoCIw\?e\=MbSmM2\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB", 'unzip': True, 'extra': False},
{'filename': "HRSOD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/Ea6kf6Kk8fpIs15WWDfJMoYBeQUeo9WXvYx9oM5yWFE1Jg\?e\=RNN0Ns\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB", 'unzip': True, 'extra': False},
{'filename': "UHRSD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EVJLvAP3HwtHksZMUolIfCABHqP7GgAWcG_1V5T_Xrnr2g\?e\=ct3pzo\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB", 'unzip': True, 'extra': False},
{'filename': "DUTS-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EbaXjDWFb6lGp9x5ae9mJPgBt9dkmgclq9XrXDjj4B5qSw\?e\=57wnhE\&download&=1", 'dest': "snapshots/InSPyReNet_SwinB_DH_LR", 'unzip': True, 'extra': True},
{'filename': "DUT-OMRON.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EU7uNIpKPXZHrDNt3gapGfsBaUrkCj67-Paj4w_7E8xs1g\?e\=n0QBR1\&download&=1", 'dest': "snapshots/InSPyReNet_SwinB_DH_LR", 'unzip': True, 'extra': True},
{'filename': "ECSSD.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EeBq5uU02FRLiW-0g-mQ_CsB3iHlMgAyOSY2Deu5suo9pQ\?e\=zjhB33\&download&=1", 'dest': "snapshots/InSPyReNet_SwinB_DH_LR", 'unzip': True, 'extra': True},
{'filename': "HKU-IS.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EY5AppwMB4hFqbISFL_u5QMBeux7dQWtDeXaMMcAyLqLqQ\?e\=N71XVt\&download&=1", 'dest': "snapshots/InSPyReNet_SwinB_DH_LR", 'unzip': True, 'extra': True},
{'filename': "PASCAL-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EfhejUtLallLroU6jjgDl-oBSHITWnkdiU6NVV95DO5YqQ\?e\=T6rrRW\&download&=1", 'dest': "snapshots/InSPyReNet_SwinB_DH_LR", 'unzip': True, 'extra': True},
{'filename': "DAVIS-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EesN4fRAE4JJk1aZ_tH3EBQBgirysALcgfw1Ipsa9dLe9Q\?e\=b5oWsg\&download&=1", 'dest': "snapshots/InSPyReNet_SwinB_DH_LR", 'unzip': True, 'extra': True},
{'filename': "HRSOD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ETpVYDijnZhInN9zBHRuwhUBSPslqe9FP0m3Eo3TWS0d5A\?e\=QTzfx6\&download&=1", 'dest': "snapshots/InSPyReNet_SwinB_DH_LR", 'unzip': True, 'extra': True},
{'filename': "UHRSD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EUkKVH3LSyFLq6UXuvWEVuYBEH8W8uAKgyolLRVuIUILag\?e\=y1SceD\&download&=1", 'dest': "snapshots/InSPyReNet_SwinB_DH_LR", 'unzip': True, 'extra': True},
{'filename': "DUTS-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EUSblgWAwg9Plc4LCGj4TLwB-7HLEdZGJqF1jHOU55g3OA\?e\=2YT3zM\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU_LR", 'unzip': True, 'extra': True},
{'filename': "DUT-OMRON.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ESNaVX4Fh5JHn5jOfgnSWi4Bx1bc9t6pg79IoG3mrpZpAw\?e\=M8D0CM\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU_LR", 'unzip': True, 'extra': True},
{'filename': "ECSSD.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ETKjTH1vcVZDu5ahRaw4cb8B7JKaPMR-0Uae1DbwarobIA\?e\=Qw67IZ\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU_LR", 'unzip': True, 'extra': True},
{'filename': "HKU-IS.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EcC9-lPgAXZHs7th9DiVjygB-zPIq_1Ii6i1GpbLGc1iPQ\?e\=EVXKp9\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU_LR", 'unzip': True, 'extra': True},
{'filename': "PASCAL-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EXjyAS4XyzlPqrpaizgElioBdmgd4E81qQzj11Qm4xo5sA\?e\=hoOzc2\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU_LR", 'unzip': True, 'extra': True},
{'filename': "DAVIS-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EcSCMR033GVNotOilIzYhIsBikzb8ZzGlkuW6aSNMlUpqQ\?e\=TFcgvE\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU_LR", 'unzip': True, 'extra': True},
{'filename': "HRSOD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EUHRSEANcmVFjcS_K-PeYr0B6VPXPgb2AHFUnlJYrf3dOQ\?e\=unwcqV\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU_LR", 'unzip': True, 'extra': True},
{'filename': "UHRSD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EU2gb0hS5kZBgKNXWqQAomsBXU-zjGCXKAzYNNk4d6EAiQ\?e\=pjhiN2\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU_LR", 'unzip': True, 'extra': True},
{'filename': "DUTS-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/Ee7Y647ERzxMgBoFceEEO6kBIUkIlmYHoizMj71gT37sxw\?e\=xDt83C\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DHU_LR", 'unzip': True, 'extra': True},
{'filename': "DUT-OMRON.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ESFGN3FCdzRAvlsW6bEaGdoBYNoJgK4DAjaS6WkVVyI_QQ\?e\=nYHklV\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DHU_LR", 'unzip': True, 'extra': True},
{'filename': "ECSSD.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EeONq5kOirRCkErrb6fFqd8B4w4SMZXBY1Q2mJvZcRsGdQ\?e\=K7fwQt\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DHU_LR", 'unzip': True, 'extra': True},
{'filename': "HKU-IS.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/Ea5P4CBBQatPiUzsH53lckoB0k23haePzuERBfyJXaCbBg\?e\=AZ96mc\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DHU_LR", 'unzip': True, 'extra': True},
{'filename': "PASCAL-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/Ea2pezB7eo1BraeBpIA8YZoBkVY38rRa3KrwSIzY1cn2dQ\?e\=o121S6\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DHU_LR", 'unzip': True, 'extra': True},
{'filename': "DAVIS-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EfNDzH8O54pGtnAxit_hUjUBK9poVq4sxxnJjSG7PUQCkw\?e\=OWt7k8\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DHU_LR", 'unzip': True, 'extra': True},
{'filename': "HRSOD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ESfZaII0pO9IqqL1FkpjIuAB8SGxLcslJeWTuKQxPNFIVA\?e\=Ce1CWg\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DHU_LR", 'unzip': True, 'extra': True},
{'filename': "UHRSD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EYsQqf1GKShJhBnkZ6gD5PABPOcjRcUGSfvTbe-wYh2O2Q\?e\=t4Xlxv\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DHU_LR", 'unzip': True, 'extra': True},
{'filename': "DUTS-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EWNaRqtzhtNFhMVfLcoyfqQBw35M8q8bxME3yZyhkTtc7Q\?e\=jrJe3v\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DH", 'unzip': True, 'extra': True},
{'filename': "DUT-OMRON.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EZF5_s8JfR9HqGBUZHSM_j4BVVMONp38_gJ1ekEdvlM-qQ\?e\=0chMdl\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DH", 'unzip': True, 'extra': True},
{'filename': "ECSSD.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EZuaFObyNOtKg0W5cM7bqPYBZYGg7Z3V3i4sClI6bU_ntA\?e\=BxxQI7\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DH", 'unzip': True, 'extra': True},
{'filename': "HKU-IS.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EY-yEnNNNT5KpiURhiDAkDEBMkiA1QwQ_T0wB1UC75GXVg\?e\=Lle02B\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DH", 'unzip': True, 'extra': True},
{'filename': "PASCAL-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EYfbsgWhm7lAlX_wj_WZZowBV_-l-UvvThC4LJEKpV0BQQ\?e\=zTiKpI\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DH", 'unzip': True, 'extra': True},
{'filename': "DAVIS-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/Ef0GUP7c0bBBonHlqgB988YB0rgxCFq3oo0u8xCN8wfyyQ\?e\=LCb8UV\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DH", 'unzip': True, 'extra': True},
{'filename': "HRSOD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EVUZBnRpa35AmrvdUybsQDMBzMZvuJWe5tT7635lh9MHDQ\?e\=FlpQW1\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DH", 'unzip': True, 'extra': True},
{'filename': "UHRSD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ETfZ_zdrDvhOh21u2mqVhigBSxn3vlfKVIwXhRfzzSSFzA\?e\=kXBBi9\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DH", 'unzip': True, 'extra': True},
{'filename': "DUTS-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EZq4JUACKCBMk2bn4yoWz6sBOKrSFTPfL7d5xopc1uDw_A\?e\=RtVHSl\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU", 'unzip': True, 'extra': True},
{'filename': "DUT-OMRON.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ETJaqoSPaYtNkc8eSGDeKzMBbjbuOAWgJwG4q52bW87aew\?e\=Pguh4b\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU", 'unzip': True, 'extra': True},
{'filename': "ECSSD.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EZAeCI6BPMdNsicnQ-m1pVEBwAhOiIcbelhOMoRGXGEvVA\?e\=BQKd7Q\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU", 'unzip': True, 'extra': True},
{'filename': "HKU-IS.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EVmGvZGz54JOvrIymLsSwq4Bpos3vWSXZm3oV7-qmGZgHA\?e\=4UhDgv\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU", 'unzip': True, 'extra': True},
{'filename': "PASCAL-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ERHDUybOh4ZKkqWZpcu7MiMBFuTK6wACkKUZaNeEQGbCNQ\?e\=GCQnoe\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU", 'unzip': True, 'extra': True},
{'filename': "DAVIS-S.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ESPmZXTnfO5CrCoo_0OADxgBt_3FoU5mSFoSE4QWbWxumQ\?e\=HAsAYz\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU", 'unzip': True, 'extra': True},
{'filename': "HRSOD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EdTnUwEeMZNBrPSBBbGZKQcBmVshSTfca9qz_BqNpAUpOg\?e\=HsJ4Gx\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU", 'unzip': True, 'extra': True},
{'filename': "UHRSD-TE.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ET48owfVQEdImrh0V4gx8_ABsYXgbIJqtpq77aK_U28VwQ\?e\=h8er3H\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_HU", 'unzip': True, 'extra': True},
{'filename': "DIS-VD.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EUbzddb_QRRCtnXC8Xl6vZoBC6IqOfom52BWbzOYk-b2Ow\?e\=aqJYi1\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DIS5K_LR", 'unzip': True, 'extra': True},
{'filename': "DIS-TE1.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ESeW_SOD26tHjBLymmgFaXwBIJlljzNycaGWXLpOp_d_kA\?e\=2EyMai\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DIS5K_LR", 'unzip': True, 'extra': True},
{'filename': "DIS-TE2.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EYWT5fZDjI5Bn-lr-iQM1TsB1num0-UqfJC1TIv-LuOXoA\?e\=jCcnty\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DIS5K_LR", 'unzip': True, 'extra': True},
{'filename': "DIS-TE3.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EQXm1DEBfaNJmH0B-A3o23kBn4v5j53kP2nF9CpG9SQkyw\?e\=lEUiZh\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DIS5K_LR", 'unzip': True, 'extra': True},
{'filename': "DIS-TE4.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EZeH2ufGsFZIoUh6D8Rtv88BBF_ddQXav4xYXXRP_ayEAg\?e\=AMzIp8\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DIS5K_LR", 'unzip': True, 'extra': True},
{'filename': "DIS-VD.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EUisiphB_W5BjgfpIYu9oNgB_fY4XxL-MhR2gR-ZZUt49Q\?e\=gqorYs\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DIS5K", 'unzip': True, 'extra': True},
{'filename': "DIS-TE1.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EW2y54ROYIZFlEq5ilRFwOsBSrIm2-HGsUHPHykaJvUBfA\?e\=797fxr\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DIS5K", 'unzip': True, 'extra': True},
{'filename': "DIS-TE2.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/ER6yEGZkgWVOsL-mauYgPyoBDIoU0Mck-twEBgQi5g3Mxw\?e\=0yJZTT\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DIS5K", 'unzip': True, 'extra': True},
{'filename': "DIS-TE3.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/Edvzj0iZ8hdDthm4Q-p2YHgBhP1X5z4AAccAoUasr2nihA\?e\=1dognG\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DIS5K", 'unzip': True, 'extra': True},
{'filename': "DIS-TE4.zip", 'url': "https://postechackr-my.sharepoint.com/:u:/g/personal/taehoon1018_postech_ac_kr/EWwMORYg8DlCgGGqOQFThZ8BgIU9wV9-0DwLrKldQl7N8w\?e\=nCwuqy\&download\=1", 'dest': "snapshots/InSPyReNet_SwinB_DIS5K", 'unzip': True, 'extra': True},
]
if __name__ == '__main__':
args = _args()
if args.dest is not None:
# os.system('ln -s ' + os.path.join(args.dest, 'data', 'Train_Dataset') + ' ' + os.path.join(repopath, 'data'))
# os.system('ln -s ' + os.path.join(args.dest, 'data', 'Test_Dataset') + ' ' + os.path.join(repopath, 'data'))
os.system('ln -s ' + os.path.join(args.dest, 'data', 'backbone_ckpt') + ' ' + os.path.join(repopath, 'data'))
os.system('ln -s ' + os.path.join(args.dest, 'snapshots') + ' ' + repopath)
else:
args.dest = repopath
# os.makedirs(os.path.join(args.dest, 'data', 'Train_Dataset'), exist_ok=True)
# os.makedirs(os.path.join(args.dest, 'data', 'Test_Dataset'), exist_ok=True)
os.makedirs(os.path.join(args.dest, 'data', 'backbone_ckpt'), exist_ok=True)
os.makedirs(os.path.join(args.dest, 'snapshots'), exist_ok=True)
download_list = backbone_ckpts + pretrained_ckpts
for dinfo in download_list:
dinfo['dest'] = os.path.join(args.dest, dinfo['dest'])
if not dinfo['extra'] or args.extra:
download_and_unzip(**dinfo) |