Add small, medium and large vehicle test datasets

2020-05-11 01:34:19 -05:00 · 2020-05-11 01:34:19 -05:00 · 640c9bfc97
parent e059b751a6
commit 640c9bfc97
1 changed files with 1 additions and 1 deletions
--- a/fastreid/data/datasets/veri.py
+++ b/fastreid/data/datasets/veri.py
@ -27,7 +27,7 @@ class VeRi(ImageDataset):
    """
    dataset_dir = 'veri'

-    def __init__(self, root='/home/liuxinchen3/notespace/data', **kwargs):
+    def __init__(self, root='datasets', **kwargs):
        self.dataset_dir = osp.join(root, self.dataset_dir)

        self.train_dir = osp.join(self.dataset_dir, 'image_train')