mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update model url

pull/224/head
liaoxingyu 2020-07-31 14:59:37 +08:00
parent 2430b8ed75
commit 977aa4a357
3 changed files with 19 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
MODEL_ZOO.md
View File

@ -69,7 +69,7 @@ Bag of Specials(BoS):
| :---: | :---: | :---: |:---: | :---: |:---:|
| [SBS(R50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/Market1501/sbs_R50.yml) | ImageNet | 95.4% | 88.2% | 64.8% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/market_sbs_R50.pth) |
| [SBS(R50-ibn)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/Market1501/sbs_R50-ibn.yml) | ImageNet | 95.7% | 89.3% | 67.5% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/market_sbs_R50-ibn.pth) |
| [SBS(S50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/Market1501/sbs_S50.yml) | ImageNet | 95.0% | 87.0% | 60.6% | - |
| [SBS(S50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/Market1501/sbs_S50.yml) | ImageNet | 95.8% | 89.4% | 67.6% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/market_sbs_S50.pth) |
| [SBS(R101-ibn)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/Market1501/sbs_R101-ibn.yml) | ImageNet | 96.3% | 90.3% | 70.0% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/market_sbs_R101-ibn.pth) |
**MGN**:
@ -95,7 +95,7 @@ Bag of Specials(BoS):
| :---: | :---: | :---: |:---: | :---: | :---:|
| [AGW(R50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/DukeMTMC/AGW_R50.yml) | ImageNet | 89.0% | 79.9% | 46.1% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/duke_agw_R50.pth) |
| [AGW(R50-ibn)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/DukeMTMC/AGW_R50-ibn.yml) | ImageNet | 90.5% | 80.8% | 47.6% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/duke_agw_R50-ibn.pth) |
| [AGW(S50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/DukeMTMC/AGW_S50.yml) | ImageNet | 89.9% | 79.7% | 44.2% | - |
| [AGW(S50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/DukeMTMC/AGW_S50.yml) | ImageNet | 90.9% | 82.4% | 49.2% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/duke_agw_S50.pth) |
| [AGW(R101-ibn)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/DukeMTMC/AGW_R101-ibn.yml) | ImageNet | 91.7% | 82.3% | 50.0% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/duke_agw_R101-ibn.pth) |
**SBS**:
@ -104,7 +104,7 @@ Bag of Specials(BoS):
| :---: | :---: | :---: |:---: | :---: | :---:|
| [SBS(R50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/DukeMTMC/sbs_R50.yml) | ImageNet | 90.3% | 80.3% | 46.5% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/duke_sbs_R50.pth) |
| [SBS(R50-ibn)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/DukeMTMC/sbs_R50-ibn.yml) | ImageNet | 90.8% | 81.2% | 47.0% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/duke_sbs_R50-ibn.pth) |
| [SBS(S50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/DukeMTMC/sbs_S50.yml) | ImageNet | 90.5% | 79.1% | 42.7% | - |
| [SBS(S50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/DukeMTMC/sbs_S50.yml) | ImageNet | 91.0% | 81.4% | 47.6% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/duke_sbs_S50.pth) |
| [SBS(R101-ibn)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/DukeMTMC/sbs_R101-ibn.yml) | ImageNet | 91.9% | 83.6% | 51.5% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/duke_sbs_R101-ibn.pth) |
**MGN**:
@ -121,7 +121,7 @@ Bag of Specials(BoS):
| :---: | :---: | :---: |:---: | :---: | :---:|
| [BoT(R50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/MSMT17/bagtricks_R50.yml) | ImageNet | 74.1% | 50.2% | 10.4% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/msmt_bot_R50.pth) |
| [BoT(R50-ibn)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/MSMT17/bagtricks_R50-ibn.yml) | ImageNet | 77.0% | 54.4% | 12.5% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/msmt_bot_R50-ibn.pth) |
| [BoT(S50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/MSMT17/bagtricks_S50.yml) | ImageNet | 80.4% | 59.2% | 15.9% | - |
| [BoT(S50)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/MSMT17/bagtricks_S50.yml) | ImageNet | 80.8% | 59.9% | 16.3% | [-](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/msmt_bot_S50.pth) |
| [BoT(R101-ibn)](https://github.com/JDAI-CV/fast-reid/blob/master/configs/MSMT17/bagtricks_R101-ibn.yml) | ImageNet| 81.0% | 59.4% | 15.6% | [model](https://github.com/JDAI-CV/fast-reid/releases/download/v0.1.1/msmt_bot_R101-ibn.pth) |
**AGW**:

12
fastreid/data/datasets/vehicleid.py
View File

@ -88,8 +88,8 @@ class SmallVehicleID(VehicleID):
"""
def __init__(self, root='datasets', **kwargs):
self.dataset_dir = osp.join(root, self.dataset_dir)
self.test_list = osp.join(self.dataset_dir, 'train_test_split/test_list_800.txt')
dataset_dir = osp.join(root, self.dataset_dir)
self.test_list = osp.join(dataset_dir, 'train_test_split/test_list_800.txt')
super(SmallVehicleID, self).__init__(root, self.test_list, **kwargs)
@ -103,8 +103,8 @@ class MediumVehicleID(VehicleID):
"""
def __init__(self, root='datasets', **kwargs):
self.dataset_dir = osp.join(root, self.dataset_dir)
self.test_list = osp.join(self.dataset_dir, 'train_test_split/test_list_1600.txt')
dataset_dir = osp.join(root, self.dataset_dir)
self.test_list = osp.join(dataset_dir, 'train_test_split/test_list_1600.txt')
super(MediumVehicleID, self).__init__(root, self.test_list, **kwargs)
@ -118,7 +118,7 @@ class LargeVehicleID(VehicleID):
"""
def __init__(self, root='datasets', **kwargs):
self.dataset_dir = osp.join(root, self.dataset_dir)
self.test_list = osp.join(self.dataset_dir, 'train_test_split/test_list_2400.txt')
dataset_dir = osp.join(root, self.dataset_dir)
self.test_list = osp.join(dataset_dir, 'train_test_split/test_list_2400.txt')
super(LargeVehicleID, self).__init__(root, self.test_list, **kwargs)

18
fastreid/data/datasets/veriwild.py
View File

@ -99,9 +99,9 @@ class SmallVeRiWild(VeRiWild):
"""
def __init__(self, root='datasets', **kwargs):
self.dataset_dir = osp.join(root, self.dataset_dir)
self.query_list = osp.join(self.dataset_dir, 'train_test_split/test_3000_query.txt')
self.gallery_list = osp.join(self.dataset_dir, 'train_test_split/test_3000.txt')
dataset_dir = osp.join(root, self.dataset_dir)
self.query_list = osp.join(dataset_dir, 'train_test_split/test_3000_query.txt')
self.gallery_list = osp.join(dataset_dir, 'train_test_split/test_3000.txt')
super(SmallVeRiWild, self).__init__(root, self.query_list, self.gallery_list, **kwargs)
@ -115,9 +115,9 @@ class MediumVeRiWild(VeRiWild):
"""
def __init__(self, root='datasets', **kwargs):
self.dataset_dir = osp.join(root, self.dataset_dir)
self.query_list = osp.join(self.dataset_dir, 'train_test_split/test_5000_query.txt')
self.gallery_list = osp.join(self.dataset_dir, 'train_test_split/test_5000.txt')
dataset_dir = osp.join(root, self.dataset_dir)
self.query_list = osp.join(dataset_dir, 'train_test_split/test_5000_query.txt')
self.gallery_list = osp.join(dataset_dir, 'train_test_split/test_5000.txt')
super(MediumVeRiWild, self).__init__(root, self.query_list, self.gallery_list, **kwargs)
@ -131,8 +131,8 @@ class LargeVeRiWild(VeRiWild):
"""
def __init__(self, root='datasets', **kwargs):
self.dataset_dir = osp.join(root, self.dataset_dir)
self.query_list = osp.join(self.dataset_dir, 'train_test_split/test_10000_query.txt')
self.gallery_list = osp.join(self.dataset_dir, 'train_test_split/test_10000.txt')
dataset_dir = osp.join(root, self.dataset_dir)
self.query_list = osp.join(dataset_dir, 'train_test_split/test_10000_query.txt')
self.gallery_list = osp.join(dataset_dir, 'train_test_split/test_10000.txt')
super(LargeVeRiWild, self).__init__(root, self.query_list, self.gallery_list, **kwargs)