Update README.md

Author: Yu-Wu

README.md

@@ -10,16 +10,16 @@ This code is based on the [Open-ReID](https://github.com/Cysu/open-reid) library
 - h5py, scikit-learn, metric-learn, tqdm
 
 ### Download datasets 
-- DukeMTMC-VideoReID: [[Google Drive]](https://drive.google.com/open?id=1T5bmWetLSvLjR30hAp8S54G2ERs81Pkg) [[BaiduYun]](https://pan.baidu.com/s/1axx55Z7XDzc95i0yGr4ItQ) [[Direct Link]](http://45.62.126.139/dukemtmc_videoReID.zip). [This repo](https://github.com/Yu-Wu/DukeMTMC-VideoReID) includes more details and baseline code.
-- MARS: [[Google Drive]](https://drive.google.com/open?id=1m6yLgtQdhb6pLCcb6_m7sj0LLBRvkDW0).
+- DukeMTMC-VideoReID: [[Direct Link]](http://vision.cs.duke.edu/DukeMTMC/data/misc/DukeMTMC-VideoReID.zip)  [[Google Drive]](https://drive.google.com/open?id=1WVjJ7PwhakF40a-BgOs1Jr_a17O38eOz)  [[BaiduYun]](https://pan.baidu.com/s/1Y_unlSqZqSdU3SeBqQmE5A). This [page](https://github.com/Yu-Wu/DukeMTMC-VideoReID) contains more details and baseline code.
+- MARS: [[Google Drive]](https://drive.google.com/open?id=1m6yLgtQdhb6pLCcb6_m7sj0LLBRvkDW0)   [[BaiduYun]](https://pan.baidu.com/s/1mByTdvXFsmobXOXBEkIWFw).
 - Move the downloaded zip files to `./data/` and unzip here.
 
 
 ## Train
 
 For the DukeMTMC-VideoReID dataset:
 ```shell
-python3 run.py --dataset dukemtmc_videoReID --logs_dir logs/dukemtmc_videoReID_EF_10/ --EF 10 --mode Dissimilarity --max_frames 900
+python3 run.py --dataset DukeMTMC-VideoReID --logs_dir logs/DukeMTMC-VideoReID_EF_10/ --EF 10 --mode Dissimilarity --max_frames 900
 ```
 
 For the MARS datasaet:
@@ -53,3 +53,4 @@ If you have further questions about this paper, please do not hesitate to contac
 
 [Yu Wu's Homepage](https://yu-wu.net)
 
+

examples/oneshot_dukemtmc_videoReID_used_in_paper.pickle renamed to examples/oneshot_DukeMTMC-VideoReID_used_in_paper.pickle

@@ -7,7 +7,7 @@
 
 __factory = {
  'mars': Mars,
- 'dukemtmc_videoReID': DukeMTMC_VideoReID
+ 'DukeMTMC-VideoReID': DukeMTMC_VideoReID
 }
 
 

reid/datasets/__init__.py

@@ -9,7 +9,7 @@
 class DukeMTMC_VideoReID(Dataset):
  def __init__(self, root, split_id=0, num_val=100, download=True):
  super(self.__class__, self).__init__(root, split_id=split_id)
- self.name="dukemtmc_videoReID"
+ self.name="DukeMTMC-VideoReID"
  self.num_cams = 8
 
  if download:
@@ -67,9 +67,9 @@ def register(subdir):
  return pids, vids
 
  print("begin to preprocess mars dataset")
- trainval_pids, _ = register('train_split')
- gallery_pids, gallery_vids = register('gallery_split')
- query_pids, query_vids = register('query_split')
+ trainval_pids, _ = register('train')
+ gallery_pids, gallery_vids = register('gallery')
+ query_pids, query_vids = register('query')
  #assert query_pids <= gallery_pids
  assert trainval_pids.isdisjoint(gallery_pids)
 

reid/datasets/dukemtmc_videoReID.py

@@ -1,5 +1,5 @@
 # run DukeMTMC-VideoReID
-python3 run.py --dataset dukemtmc_videoReID --logs_dir logs/dukemtmc_videoReID_EF_10/ --EF 10 --mode Dissimilarity --max_frames 900
+python3 run.py --dataset DukeMTMC-VideoReID --logs_dir logs/DukeMTMC-VideoReID_EF_10/ --EF 10 --mode Dissimilarity --max_frames 900
 
 # run mars
 #python3 run.py --dataset mars --logs_dir logs/mars_EF_10/ --EF 10 --mode Dissimilarity --max_frames 900