k20hcmus/FishEye8K

1__model_types = [2    'resnet50', 'mlfn', 'hacnn', 'mobilenetv2_x1_0', 'mobilenetv2_x1_4',3    'osnet_x1_0', 'osnet_x0_75', 'osnet_x0_5', 'osnet_x0_25',4    'osnet_ibn_x1_0', 'osnet_ain_x1_0']5 6__trained_urls = {7 8    # market1501 models ########################################################9    'resnet50_market1501.pt':10    'https://drive.google.com/uc?id=1dUUZ4rHDWohmsQXCRe2C_HbYkzz94iBV',11    'resnet50_dukemtmcreid.pt':12    'https://drive.google.com/uc?id=17ymnLglnc64NRvGOitY3BqMRS9UWd1wg',13    'resnet50_msmt17.pt':14    'https://drive.google.com/uc?id=1ep7RypVDOthCRIAqDnn4_N-UhkkFHJsj',15 16    'resnet50_fc512_market1501.pt':17    'https://drive.google.com/uc?id=1kv8l5laX_YCdIGVCetjlNdzKIA3NvsSt',18    'resnet50_fc512_dukemtmcreid.pt':19    'https://drive.google.com/uc?id=13QN8Mp3XH81GK4BPGXobKHKyTGH50Rtx',20    'resnet50_fc512_msmt17.pt':21    'https://drive.google.com/uc?id=1fDJLcz4O5wxNSUvImIIjoaIF9u1Rwaud',22 23    'mlfn_market1501.pt':24    'https://drive.google.com/uc?id=1wXcvhA_b1kpDfrt9s2Pma-MHxtj9pmvS',25    'mlfn_dukemtmcreid.pt':26    'https://drive.google.com/uc?id=1rExgrTNb0VCIcOnXfMsbwSUW1h2L1Bum',27    'mlfn_msmt17.pt':28    'https://drive.google.com/uc?id=18JzsZlJb3Wm7irCbZbZ07TN4IFKvR6p-',29 30    'hacnn_market1501.pt':31    'https://drive.google.com/uc?id=1LRKIQduThwGxMDQMiVkTScBwR7WidmYF',32    'hacnn_dukemtmcreid.pt':33    'https://drive.google.com/uc?id=1zNm6tP4ozFUCUQ7Sv1Z98EAJWXJEhtYH',34    'hacnn_msmt17.pt':35    'https://drive.google.com/uc?id=1MsKRtPM5WJ3_Tk2xC0aGOO7pM3VaFDNZ',36 37    'mobilenetv2_x1_0_market1501.pt':38    'https://drive.google.com/uc?id=18DgHC2ZJkjekVoqBWszD8_Xiikz-fewp',39    'mobilenetv2_x1_0_dukemtmcreid.pt':40    'https://drive.google.com/uc?id=1q1WU2FETRJ3BXcpVtfJUuqq4z3psetds',41    'mobilenetv2_x1_0_msmt17.pt':42    'https://drive.google.com/uc?id=1j50Hv14NOUAg7ZeB3frzfX-WYLi7SrhZ',43 44    'mobilenetv2_x1_4_market1501.pt':45    'https://drive.google.com/uc?id=1t6JCqphJG-fwwPVkRLmGGyEBhGOf2GO5',46    'mobilenetv2_x1_4_dukemtmcreid.pt':47    'https://drive.google.com/uc?id=12uD5FeVqLg9-AFDju2L7SQxjmPb4zpBN',48    'mobilenetv2_x1_4_msmt17.pt':49    'https://drive.google.com/uc?id=1ZY5P2Zgm-3RbDpbXM0kIBMPvspeNIbXz',50 51    'osnet_x1_0_market1501.pt':52    'https://drive.google.com/uc?id=1vduhq5DpN2q1g4fYEZfPI17MJeh9qyrA',53    'osnet_x1_0_dukemtmcreid.pt':54    'https://drive.google.com/uc?id=1QZO_4sNf4hdOKKKzKc-TZU9WW1v6zQbq',55    'osnet_x1_0_msmt17.pt':56    'https://drive.google.com/uc?id=112EMUfBPYeYg70w-syK6V6Mx8-Qb9Q1M',57 58    'osnet_x0_75_market1501.pt':59    'https://drive.google.com/uc?id=1ozRaDSQw_EQ8_93OUmjDbvLXw9TnfPer',60    'osnet_x0_75_dukemtmcreid.pt':61    'https://drive.google.com/uc?id=1IE3KRaTPp4OUa6PGTFL_d5_KQSJbP0Or',62    'osnet_x0_75_msmt17.pt':63    'https://drive.google.com/uc?id=1QEGO6WnJ-BmUzVPd3q9NoaO_GsPNlmWc',64 65    'osnet_x0_5_market1501.pt':66    'https://drive.google.com/uc?id=1PLB9rgqrUM7blWrg4QlprCuPT7ILYGKT',67    'osnet_x0_5_dukemtmcreid.pt':68    'https://drive.google.com/uc?id=1KoUVqmiST175hnkALg9XuTi1oYpqcyTu',69    'osnet_x0_5_msmt17.pt':70    'https://drive.google.com/uc?id=1UT3AxIaDvS2PdxzZmbkLmjtiqq7AIKCv',71 72    'osnet_x0_25_market1501.pt':73    'https://drive.google.com/uc?id=1z1UghYvOTtjx7kEoRfmqSMu-z62J6MAj',74    'osnet_x0_25_dukemtmcreid.pt':75    'https://drive.google.com/uc?id=1eumrtiXT4NOspjyEV4j8cHmlOaaCGk5l',76    'osnet_x0_25_msmt17.pt':77    'https://drive.google.com/uc?id=1sSwXSUlj4_tHZequ_iZ8w_Jh0VaRQMqF',78 79    ####### market1501 models ##################################################80    'resnet50_msmt17.pt':81    'https://drive.google.com/uc?id=1yiBteqgIZoOeywE8AhGmEQl7FTVwrQmf',82    'osnet_x1_0_msmt17.pt':83    'https://drive.google.com/uc?id=1IosIFlLiulGIjwW3H8uMRmx3MzPwf86x',84    'osnet_x0_75_msmt17.pt':85    'https://drive.google.com/uc?id=1fhjSS_7SUGCioIf2SWXaRGPqIY9j7-uw',86 87    'osnet_x0_5_msmt17.pt':88    'https://drive.google.com/uc?id=1DHgmb6XV4fwG3n-CnCM0zdL9nMsZ9_RF',89    'osnet_x0_25_msmt17.pt':90    'https://drive.google.com/uc?id=1Kkx2zW89jq_NETu4u42CFZTMVD5Hwm6e',91    'osnet_ibn_x1_0_msmt17.pt':92    'https://drive.google.com/uc?id=1q3Sj2ii34NlfxA4LvmHdWO_75NDRmECJ',93    'osnet_ain_x1_0_msmt17.pt':94    'https://drive.google.com/uc?id=1SigwBE6mPdqiJMqhuIY4aqC7--5CsMal',95}96 97 98def show_downloadeable_models():99    print('\nAvailable .pt ReID models for automatic download')100    print(list(__trained_urls.keys()))101 102 103def get_model_url(model):104    model = str(model).rsplit('/', 1)[-1]105    if model in __trained_urls:106        return __trained_urls[model]107    else:108        None109 110 111def is_model_in_model_types(model):112    model = str(model).rsplit('/', 1)[-1].split('.')[0]113    if model in __model_types:114        return True115    else:116        return False117 118 119def get_model_name(model):120    model = str(model).rsplit('/', 1)[-1].split('.')[0]121    for x in __model_types:122        if x in model:123            return x124    return None125 126